我有一个类似于这个的大熊猫框架:
import pandas as pd
import numpy as np
data = {'Col1' : [4,5,6,7], 'Col2' : [10,20,30,40], 'Col3' : [100,50,-30,-50], 'Col4' : ['AAA', 'BBB', 'AAA', 'CCC']}
df = pd.DataFrame(data=data, index = ['R1','R2','R3','R4'])
Col1 Col2 Col3 Col4
R1 4 10 100 AAA
R2 5 20 50 BBB
R3 6 30 -30 AAA
R4 7 40 -50 CCC
给定一系列目标:
target_array = np.array(['AAA', 'CCC', 'EEE'])
我想在Col4中找到也出现在target_array中的单元格索引.
我试图找到一个记录的答案,但似乎超出了我的技能…任何人有任何建议?
附:顺便说一下,对于这种特殊情况,我可以输入一个目标数组,其元素是数据帧索引名称数组([‘R1′,’R3′,’R5’]).这样会更容易吗?
编辑1:
非常感谢所有伟大的回复.可悲的是,我只能选择一个,但每个人似乎都把@Divakar当成最好的.您仍然应该查看piRSquared和MaxU的速度比较,以获得所有可能的可能性