python - 根据 numpy 数组中的列值获取行号

Question

我是 numpy 新手，需要一些帮助来解决我的问题。我使用 dtypes 从二进制文件中读取记录，然后选择 3 列

df = pd.DataFrame(np.array([(124,90,5),(125,90,5),(126,90,5),(127,90,0),(128,91,5),(129,91,5),(130,91,5),(131,91,0)]), columns =  ['atype','btype','ctype'] )

这给出了

   atype    btype   ctype
0    124       90       5
1    125       90       5
2    126       90       5
3    127       90       0
4    128       91       5
5    129       91       5
6    130       91       5
7    131       91       0

我现在对“atype”不感兴趣。但我想要的是当(x,90,5) 出现在第二列和第三列(x,90,0) 出现在第二列和第三列当 (x,91,5) 出现在第二列和第三列时(x,91,0) 出现在第二列和第三列等等

有 7 个变量，例如 90,91,92,93,94,95,96，相应地，第三列中会有 5 或 0 的值。

参赛作品数量为100万。那么是否有办法在不使用 for 循环的情况下找出这些内容。

Answer 1

使用 pandas 你可以尝试以下操作。

df[(df['btype'].between(90, 96)) & (df['ctype'].isin([0, 5]))]

使用你的例子。如果某些值发生更改，例如 df 为

   atype    btype   ctype
0    124       90       5
1    125       90       5
2    126        0       5
3    127       90     100
4    128       91       5
5    129        0       5
6    130       91       5
7    131       91       0

然后使用上面的解决方案，返回以下内容。

   atype    btype   ctype
0    124       90       5
1    125       90       5
4    128       91       5
6    130       91       5
7    131       91       0