在这个Python 3.11代码片段中:
import numpy as np
state = np.arange(48, dtype='u1').reshape((2, 8, 3))
pixels = [3, 4, 5]
colors = [[42, 43, 44], [0, 1, 2]]
state[0, pixels] = colors[0] # line 1
state[1, pixels] = colors[1] # line 2
# state[:, pixels, :] = colors # error
我想把line 1
和line 2
替换成一个单行的Numpy魔法。最后一行无法编译。
1条答案
按热度按时间cotxawn71#
colors
需要与state
具有相同的维数输出