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

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 dfget_choice(sample_dataframe, 'Sample')

Answer 1

我相信您在尝试将与样本中的元素对应的输入添加到新列中。

import pandas as pd

sample_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 df

print(get_choice(sample_dataframe, 'Sample'))