Я пытаюсь написать функцию, которая берет имя столбца и разбивает фрейм данных на основе значений этого столбца. у меня есть следующее
df_split = df[df.a == 1]
Я пытаюсь реализовать следующую идею
def f(df,column_name):
df_split = df[df.column_name == 1]
Любая помощь высоко ценится.
Спасибо, prahasanam_boi, все работает.
поэтому, разместив это в ответах
Пожалуйста, измените функцию на следующую:
def f(df,column_name):
df_split = df[df[column_name] == 1]
return df_split
df.column_name будет работать только в том случае, если в фрейме данных действительно есть столбец, помеченный как column_name
, поэтому не используйте его внутри функции.
df_split = df[df[column_name] == 1]
ты этого хочешь? если нет, пожалуйста, дайте больше ясности по вопросу