我有由0和1组成的数字矩阵.示例:
p=[[1 0 0 1 0]
[0 0 0 0 0]
...
[1 1 1 1 1]
[0 0 0 1 1]]
这个麻木矩阵的行表示5个元素数据突发的时间间隔测量.每列表示测量通道:
channels=[A B C D E]
我需要将频道映射到p上，这样我就得到:
q=[[A None None D None]
[None None None None None]
...
[A B C D E]
[None None None D E]]
将p映射到q的最Python方法是什么？
您可以使用np.tile的组合首先扩展字符串标签，然后使用np.where从p中 Select 特定索引或将结果设置为None.
import numpy as np
p = np.array(
[[1, 0, 0, 1, 0], [0, 0, 0, 0, 0], [1, 1, 1, 1, 1], [0, 0, 0, 1, 1]]
)
labels = ["A", "B", "C", "D", "E"]
expanded = np.tile(labels, (len(p), 1))
final = np.where(p, expanded, None)
print(final)
正如@Mechanic Pig指出的那样，广播本身就可以解决这个问题，所以你甚至不需要np.tile个.