내가 가지고 있고 mxn 배열이라고 가정합니다. 이 배열의 각 열을 함수에 전달하여 전체 열에 대해 일부 작업을 수행하고 싶습니다. 배열의 열을 어떻게 반복합니까?
예를 들어, 다음과 같은 4 x 3 배열이 있습니다.
1 99 2
2 14 5
3 12 7
4 43 1
for column in array:
some_function(column)여기서 열은 첫 번째 반복에서 “1,2,3,4”, 두 번째 반복에서 “99,14,12,43”, 세 번째 반복에서 “2,5,7,1”입니다.
답변
배열의 전치에 대해 반복하십시오.
for column in array.T:
some_function(column)답변
이것은 당신에게 시작을 줄 것입니다
>>> for col in range(arr.shape[1]):
some_function(arr[:,col])
[1 2 3 4]
[99 14 12 43]
[2 5 7 1]답변
3 차원 배열의 경우 다음을 시도 할 수 있습니다.
for c in array.transpose(1, 0, 2):
do_stuff(c)array.transpose작동 방식 에 대한 문서를 참조하십시오 . 기본적으로 이동할 차원을 지정합니다. 이 경우 두 번째 차원 (예 : 열)을 첫 번째 차원으로 이동합니다.
답변
for c in np.hsplit(array, array.shape[1]):
some_fun(c)답변
압축 해제를 사용하여 열을 반복 할 수도 있습니다.
for col in zip(*array):
some_function(col)답변
예를 들어 행렬에서 각 열의 평균을 찾고 싶습니다. 다음 행렬을 만들어 보겠습니다.
mat2 = np.array([1,5,6,7,3,0,3,5,9,10,8,0], dtype=np.float64).reshape(3, 4)평균 함수는 다음과 같습니다.
def my_mean(x):
return sum(x)/len(x)필요한 작업을 수행하고 결과를 콜론 벡터 ‘결과’에 저장하려면
results = np.zeros(4)
for i in range(0, 4):
mat2[:, i] = my_mean(mat2[:, i])
results = mat2[1,:] 결과는 다음과 같습니다. array ([4.33333333, 5., 5.66666667, 4.])
답변
또는 enumerate. 열 번호와 열 값도 제공합니다.
for num, column in enumerate(array.T):
some_function(column) # column: Gives you the column value as asked in the question
some_function(num) # num: Gives you the column number
