说我有

data=np.array([[30,0.109,1],
               [25,0.517,2],
               [22,0.174,1],
               [35,0.812,3],
               [45,0.215,4],
               [40,0.111,4],
               [50,0.095,4]])

做

A = data1[:,2]
print(A)

会给予我

[1. 2. 1. 3. 4. 4. 4.]

如果第3列的值为4，是否有命令可以从其他列（1或2）中获取值？在我的脑海中，它看起来像这样B=data[：，1，[2]=4]所以B应该打印出[0.812，0.215，0.111]
这可能吗？我需要它的原因是图形显示。第3列中的数字对应于细菌的类型（即，所有的4都是相同的细菌），我需要从中绘制与该类型细菌“绑定”的其他值。
希望这是可能的，一切顺利