Nie używając pętli:
import numpy as np
x = np.random.randint(1, 6, 10)
print(x)
Nie używając pętli i diag():
2) napisać program, który znajduje elementy macierzy, leżące na głównej diagonali:
import numpy as np
x = np.arange(100).reshape(10,10)
Sprawia mi to trudności.
Za pomocą pętli nie można, ale chyba list comprehension to nie pętla
import numpy as np
x = np.random.randint(1,6,10)
condition = [x[i] == x[i+1] for i in range(len(x)-1)]
print(np.extract(condition, x))
https://docs.scipy.org/doc/numpy/reference/generated/numpy.extract.html#numpy.extract
Można też użyć reduce:
import functools
x = [1, 2, 2, 3, 4, 5, 6, 6, 6, 7, 8, 9, 1, 1, 1, 2, 3]
def my_aggr(acc, arg):
if arg == acc[1]:
acc[0].append(arg)
else:
acc[1] = arg
return acc
# akumulator przechowuje: listę wartości, ostatnio rozważaną wartość
yy = functools.reduce(my_aggr, x, [[], None])
print(yy[0])
Jeśli oprócz wartości chcesz znać np. indeksy, to trzeba zmodyfikować funkcję redukującą.
