python - Pandas，如果值在列中，则更改为另一列

Question

这个问题已经有答案了:

Pandas conditional creation of a series/dataframe column (13 个回答)

已关闭 3 年前。

我想做:如果列包含值，则对 joinkey 列应用更改，如果列包含不同的值，则对 joinkey 列应用不同的更改，等等。

IF df.loc[df['product'].isin('value1','value2')然后df['joinkey'] = df['joinkey'] + variable1

然后重复

IF df.loc[df['product'].isin('value3','value4')然后df['joinkey'] = df['joinkey'] + variable2

希望这是有道理的，任何帮助将不胜感激。

谢谢。

Answer 1

您可以尝试对列使用np.where()，也许它需要多行代码。

import pandas as pd
import numpy as np
df['joinkey'] = np.where(df['product'].isin(['value1','value2']),df['joinkey']+variable1,df['joinkey'])
df['joinkey'] = np.where(df['product'].isin(['value3','value4']),df['joinkey']+variable2,df['joinkey'])

我已将 false 条件设置为等于 joinkey 的原始值，但当然您可以更改它以最适合您的需求。