import randomx =[[random.randint(1,10) for j in range(5)] for i in range(4)]for i in range(4): for j in range(5): print("%4d" % (x[i][j]), end="") print(end='\n')我需要在二維數(shù)組中找到每一列的最小元素并將它們添加到一個新數(shù)組（一維數(shù)組）中
4 回答
肥皂起泡泡
TA貢獻1829條經(jīng)驗 獲得超6個贊
您可以使用列表理解來應(yīng)用于min每一列數(shù)據(jù)：
import random
x =[[random.randint(1,10) for j in range(5)] for i in range(4)]
for i in range(4):
for j in range(5):
print("%4d" % (x[i][j]), end="")
print(end='\n')
print([min(x[i][j] for i in range(4)) for j in range(5)])
示例輸出：
8 1 7 6 9
9 2 8 6 8
10 5 3 3 5
1 9 7 10 9
[1, 1, 3, 3, 5]
萬千封印
TA貢獻1891條經(jīng)驗 獲得超3個贊
您應(yīng)該查看 numpy() ，特別是np.min()。然后代碼可以是：
# Convert 'x' to np array-
x_np = np.asarray(x)
# Sanity check-
x_np
'''
array([[ 8, 7, 10, 8, 8],
[ 6, 6, 3, 6, 6],
[ 9, 4, 2, 8, 3],
[ 2, 4, 5, 3, 4]])
'''
# Minimum elements by squashing the columns-
np.min(x_np, axis = 1)
# array([7, 3, 2, 2])
# Minimum elements by squashing the rows-
np.min(x_np, axis = 0)
# array([2, 4, 2, 3, 3])
添加回答
舉報
0/150
提交
取消