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

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)

我想添加用户输入，以基于"Sample"列创建列"Condition"。下面的函数返回错误"0"；TypeError:"DataFrame"对象不可调用">

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 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'))