如何根据另一列过滤列？

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。