如何使用用户输入在熊猫中创建列

Question

我有一个数据表：

sample_data = {'Sample': ['A', 'B', 'A', 'B'],

                'Surface': ['Top', 'Bottom', 'Top', 'Bottom'],

                'Intensity' : [21, 32, 14, 45]}

sample_dataframe = pd.DataFrame(data=sample_data)

我想添加用户输入以创建基于“示例”列的“条件”列。下面的函数返回错误“TypeError: 'DataFrame' object is not callable”

def get_choice(df, column):

    for i in column:

        user_input = input('Condition= ')

        df['Condition'] = df(user_input)

    return df

get_choice(sample_dataframe, 'Sample')

Answer 1

我相信您在尝试将与样本中的元素对应的输入添加到新列中。import pandas as pdsample_data = {'Sample': ['A', 'B', 'A', 'B'],                'Surface': ['Top', 'Bottom', 'Top', 'Bottom'],                'Intensity' : [21, 32, 14, 45]}sample_dataframe = pd.DataFrame(data=sample_data)def get_choice(df, column):    user_input=[]    for i in df[column]:        print(i)        user_input.append(input('Condition= '))    df['Condition'] = user_input    return dfprint(get_choice(sample_dataframe, 'Sample'))