How to use user input to create a column in pandas

Question

I have a data table:

sample_data = {'Sample': ['A', 'B', 'A', 'B'],
                'Surface': ['Top', 'Bottom', 'Top', 'Bottom'],
                'Intensity' : [21, 32, 14, 45]}

sample_dataframe = pd.DataFrame(data=sample_data)

I would like to add user input to create a column 'Condition' based on the 'Sample' column. The function below returns the error "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')

It seems weird to me that you pass a string column name and then proceed to iterate over the characters (??). Did you mean to pass ['Sample', ] instead of 'Sample'? — cs95
– cs95, Commented Jul 14, 2020 at 18:33

Adhun Thalekkara · Accepted Answer · 2020-07-14 18:43:18Z

1

i believe you where trying to add inputs corresponding to elements in sample to a new column.

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

answered Jul 14, 2020 at 18:43

Adhun Thalekkara

72310 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to use user input to create a column in pandas

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related