Есть ли способ ввести имена столбцов без кавычек и использовать их как подмножество в другой функции в pandas?

Question

У меня есть фрейм данных, где я пытаюсь выделить дубликаты, используя:

print(df.duplicated(subset = column_subset, keep=False))

Я хочу, чтобы пользователи этого скрипта вводили column_subset, используя:

column_subset = input("Enter list of columns for group by:")

Я хочу, чтобы пользователи введите имена столбцов следующим образом: val1,val2,val3

Есть ли способ сделать это?

Answer 1

Вы можете сделать что-то вроде этого:

column_subset = []

# number of elements as input 
n = int(input("Enter number of elements : ")) 

# iterating till the range 
for i in range(0, n): 
    ele = int(input()) 
    column_subset.append(ele) # adding the element 

Вы можете заставить пользователя вводить ввод один за другим и добавлять эти элементы в список. Используйте этот список в вашей дублированной команде.

Если вы хотите, чтобы пользователь вводил список имен столбцов, просто сделайте это:

column_subset = input("Enter list of columns for group by:")
column_subset = column_subset.split(',')