python - 从 pandas where() 中排除列

Question

我有以下 pandas df :

import pandas as pd
import numpy as np    

pd_df = pd.DataFrame({'Qu1': ['apple', 'potato', 'cheese', 'banana', 'cheese', 'banana', 'cheese', 'potato', 'egg'],
              'Qu2': ['sausage', 'banana', 'apple', 'apple', 'apple', np.nan, 'banana', 'banana', 'banana'],
              'Qu3': ['apple', 'potato', 'sausage', 'cheese', 'cheese', 'potato', 'cheese', 'potato', 'egg']})

我想仅在 Qu1 和 Qu2 两列上实现 where() 并保留其余部分 original stackoverflow question ，所以我创建了 pd1

pd1 = pd_df.where(pd_df.apply(lambda x: x.map(x.value_counts()))>=2,
                              "other")[['Qu1', 'Qu2']]

然后我将其余的 pd_df、pd_df['Qu3'] 添加到 pd1

pd1['Qu3'] = pd_df['Qu3']
pd_df = []

我的问题是:最初我想在 df 的一部分上执行 where() 并保持其余列不变，所以可以上面的代码对于大型数据集有危险吗？我这样会损坏原始数据吗？如果是，最好的方法是什么？

非常感谢!

Answer 1

您可以明确地获取原始 df 的副本，然后覆盖该 df 的选择:

In [40]:
pd1 = pd_df.copy()
pd1[['Qu1', 'Qu2']] = pd1[['Qu1', 'Qu2']].where(pd_df.apply(lambda x: x.map(x.value_counts()))>=2,
                              "other")
pd1

Out[40]:
      Qu1     Qu2      Qu3
0   other   other    apple
1  potato  banana   potato
2  cheese   apple  sausage
3  banana   apple   cheese
4  cheese   apple   cheese
5  banana   other   potato
6  cheese  banana   cheese
7  potato  banana   potato
8   other  banana      egg

所以这里的区别在于我们只对 df 的一部分进行操作，而不是对整个 df 进行操作，然后选择感兴趣的列

更新

如果您只想覆盖这些列，则只需选择它们:

In [48]:
pd_df[['Qu1', 'Qu2']] = pd_df[['Qu1', 'Qu2']].where(pd_df.apply(lambda x: x.map(x.value_counts()))>=2,
                              "other")
pd_df

Out[48]:
      Qu1     Qu2      Qu3
0   other   other    apple
1  potato  banana   potato
2  cheese   apple  sausage
3  banana   apple   cheese
4  cheese   apple   cheese
5  banana   other   potato
6  cheese  banana   cheese
7  potato  banana   potato
8   other  banana      egg