使用比较特定数组项努比.wh

Question

我一直在寻找某种方法来比较使用numpy.where的特定数组项。我有一个熊猫数据框如下：

   Id   City                    UF
0   1   [5057, Itu, 26]         [26, São Paulo]
1   2   [5366, Sorocaba, 26]    None
2   3   [5347, São Paulo, 26]   [26, São Paulo]
3   4   [3288, Curitiba, 18]    [18, Paraná]
4   5   [162, Manaus, 3]        [3, Amazonas]

对于每一行，我想比较arrayCity中的第三项与arrayUF中的第一项。我试过这样：

np.where((pdf.loc[pdf['Cidade']:, 2] == pdf.loc[pdf['UF']:, 0]), True, False)

但没有成功。我找到的每个类似的样本都是关于比较整个数组的。你知道吗

下面是一些我已经检查过的链接here也here

谢谢你的建议。你知道吗

Answer 1

几乎和你说的一样。你知道吗

我用下面的代码得到了我想要的：

np.where((pdf['Cidade'].apply(lambda x: x[2] if x is not None else None) == pdf['UF'].apply(lambda x: x[0] if x is not None else None)), True, False)

谢谢你的小费！你知道吗