按列迭代:
import numpy as np
matrix = np.array([[1,2,3],[4,5,6],[7,8,9]])
for col in range(matrix.shape[1]):
column = matrix[:,col]
if np.all(column > 2):
print("Column {} has all elements greater than 2".format(col))
按行迭代:
import numpy as np
matrix = np.array([[1,2,3],[4,5,6],[7,8,9]])
for row in range(matrix.shape[0]):
row = matrix[row,:]
if np.all(row > 2):
print("Row {} has all elements greater than 2".format(row))
下一篇:按列获取单词的平均值