如何过滤基于另一列的列？

Question

我有一个这样的数据框：

X    Y
A1   1
A2   2
A1   2
A3   3
A1   3
A1   4

如果Y列的范围是1到4，

• 我需要运行一个脚本，该脚本在其中返回Y的相应X值，范围为1到4。
• 例如，当Y范围为1到4时，输出应为A1。
• 如果Y为2，则输出应为A2

我试过使用这个

df = df.pivot_table(index='X', columns='Y', aggfunc='size', fill_value=0)

但是得到了1、2、3、4的列列表。

Answer 1

我假设您正在对数据帧使用Pandas。 如果是这种情况，则可以使用loc提取所需的部分数据框。 您可以尝试以下代码段：

df.loc[df['Y'] in range(1,4), 'X']

有关loc的更多信息，您可以访问docs 。