Khó làm thế nào để lặp qua các cột của mảng?

Giả sử tôi có và mxn mảng. Tôi muốn chuyển mỗi cột của mảng này cho một hàm để thực hiện một số thao tác trên toàn bộ cột. Làm cách nào để lặp qua các cột của mảng?

Ví dụ: tôi có một mảng 4 x 3 như

1  99 2
2  14 5
3  12 7
4  43 1

for column in array:
  some_function(column)

trong đó cột sẽ là "1,2,3,4" trong lần lặp đầu tiên, "99,14,12,43" trong lần thứ hai và "2,5,7,1" trong lần thứ ba.

Chỉ cần lặp lại phần đã hoán vị của mảng của bạn:

for column in array.T:
   some_function(column)

Điều này sẽ cho bạn một khởi đầu

>>> for col in range(arr.shape[1]):
    some_function(arr[:,col])


[1 2 3 4]
[99 14 12 43]
[2 5 7 1]

Đối với mảng ba chiều, bạn có thể thử:

for c in array.transpose(1, 0, 2):
    do_stuff(c)

Xem tài liệu về cách array.transposehoạt động. Về cơ bản, bạn đang chỉ định thứ nguyên nào sẽ thay đổi. Trong trường hợp này, chúng tôi đang chuyển thứ nguyên thứ hai (ví dụ: cột) sang thứ nguyên đầu tiên.

for c in np.hsplit(array, array.shape[1]):
    some_fun(c)

Bạn cũng có thể sử dụng giải nén để lặp lại các cột

for col in zip(*array):
   some_function(col)

Ví dụ, bạn muốn tìm giá trị trung bình của mỗi cột trong ma trận. Hãy tạo ma trận sau

mat2 = np.array([1,5,6,7,3,0,3,5,9,10,8,0], dtype=np.float64).reshape(3, 4)

Hàm có nghĩa là

def my_mean(x):
    return sum(x)/len(x)

Để làm những gì cần thiết và lưu trữ kết quả trong 'kết quả' vectơ dấu hai chấm

results = np.zeros(4)
for i in range(0, 4):
    mat2[:, i] = my_mean(mat2[:, i])

results = mat2[1,:]      

Kết quả là: array ([4.33333333, 5., 5.66666667, 4.])

Ngoài ra, bạn có thể sử dụng enumerate. Nó cung cấp cho bạn số cột và các giá trị cột.

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