我遇到了一個簡單的問題,我想根據表示陣列中元素位置的掩碼來分配值。例如,array[*,1] = 0但這段代碼顯然不起作用。
經過一番思考,我想出了這個:
import numpy as np
a = np.random.normal(size=(5, 2))
print(a)
print(a.shape)
for i in np.arange(a.shape[0]):
a[i][1] = 0
print(a)
print(a.shape)
但很明顯,這個笨拙的回圈不是 Pythonic 的方法。
那么,您能分享一些在 Python 中執行此類操作的巧妙方法嗎?
uj5u.com熱心網友回復:
a[:, 1] = 0如果您希望第一列的所有條目為零,只需將陣列切片。
如果你想使用條件陳述句查看 np.where,它可以用于根據 a[np.where(condition)] = 0
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/382264.html
