主页 / 问题 / 1228440
Accepted
chasesilvers
Asked:2022-01-08 05:56:12 +0800 CST2022-01-08 05:56:12 +0800 CST

如何通过列中的值列表显示DF中的行？

  • 772

我有这个 DF：

df = pd.DataFrame({'A': ['100','110','120','130','140','150','160','170'], 
'B': ['a','b','c','d','f','g','h','y'], 'C': ['a','b','c','d','f','g','h','y']})

A   B   C
0   100 a   a
1   110 b   b
2   120 c   c
3   130 d   d
4   140 f   f
5   150 g   g
6   160 h   h
7   170 y   y

还有一个我从另一个表中得到的值列表，比如说：

list = [110,140,170]

如何将这些值应用到 A 列并为它们显示相应的行？

结果应该是这样的：

    A   B   C
1   110 b   b
2   140 f   f
3   170 y   y
python

1 个回答

  1. Best Answer

    列表和列中的值必须是同一类型，并且A列中有字符串，列表中有整数。

    In [138]: s = pd.Series(list_).astype(str)

In [139]: df.query("A in @s")
Out[139]: 
     A  B  C
1  110  b  b
4  140  f  f
7  170  y  y

    反之亦然 - 我们将列转换为数字数据类型，然后查找：

    In [141]: df[df["A"].astype(int).isin(list_)]
Out[141]: 
     A  B  C
1  110  b  b
4  140  f  f
7  170  y  y

    如果在您的真实数据中，列和列表中的数据类型相同，那么解决方案看起来会更简单：

    res = df.query("A in @list_")

    或像这样：

    res = df[df["A"].isin(list_)]

    PS 不要使用保留字（如list）作为自己的变量、函数和类的名称，以免出现不必要的问题。

    • 2

相关问题