我有以下 python 代碼片段：import numpy as np# Some random arrayx = np.random.rand(34, 176, 256)# Get some indexespos_idx = np.argwhere(x > 0.5)# Sample some values from these indexesseeds = pos_idx[np.random.choice(pos_idx.shape[0], size=5, replace=False), :]# Now create another arrayy = np.zeros_like(x, np.uint8)y[seeds] = 1最后一行給出了如下錯(cuò)誤：index 77 is out of bounds for axis 0 with size 34但我不確定這是怎么發(fā)生的，因?yàn)樗胁蓸铀饕紤?yīng)該有效，因?yàn)樗鼈兪且粋€(gè)子集。
3 回答
MYYA
TA貢獻(xiàn)1868條經(jīng)驗(yàn) 獲得超4個(gè)贊
它將種子中的值視為第一維的索引。要通過種子中的索引訪問元素，您可以使用：
y[seeds[:,0],seeds[:,1],seeds[:,2]] = 1
哆啦的時(shí)光機(jī)
TA貢獻(xiàn)1779條經(jīng)驗(yàn) 獲得超6個(gè)贊
此代碼將幫助您將值設(shè)置為 1
import numpy as np
# Some random array
x = np.random.rand(34, 176, 256)
# Get some indexes
pos_idx = np.argwhere(x > 0.5)
# Sample some values from these indexes
seeds = pos_idx[np.random.choice(pos_idx.shape[0], size=5, replace=False), :]
# Now create another array
y = np.zeros_like(x, np.uint8)
for i in seeds:
y[tuple(i)] = 1
慕碼人2483693
TA貢獻(xiàn)1860條經(jīng)驗(yàn) 獲得超9個(gè)贊
種子的形狀是否正確？如果我理解正確的話，x有隨機(jī)值；y與 x 具有相同的形狀，但全為零；andseeds是一些索引值，這些位置將被設(shè)置為一個(gè)。
print('x : ', x.ndim, x.shape)
print('y : ', y.ndim, y.shape)
print('seeds: ', seeds.ndim, seeds.shape)
x : 3 (34, 176, 256)
y : 3 (34, 176, 256)
seeds: 2 (5, 3)
添加回答
舉報(bào)
0/150
提交
取消