Как написать пустой столбец в CSV на основе других столбцов в том же файле CSV

Question

Я не знаю, является ли это очень простым вопросом, но я хотел бы сделать оператор условия, основанный на двух других столбцах.У меня есть два столбца, такие как: возраст и SES и еще один пустой столбец, который должен быть основан на этих двух столбцах.Например, если одному человеку 65 лет и его соответствующий социально-экономический статус высокий, то в третьем столбце (пустой столбец = класс жизнеспособности) приводится, например, значение 1 .У меня есть представление о том, чего я хочу достичь, однако я понятия не имею, как реализовать это в самом Python.Я знаю, что должен использовать цикл для , и я знаю, как писать условия, однако из-за того, что я хочу принять во внимание два столбца для определения того, что будет написано в пустом столбце, у меня нетидея, как записать это в функцию и, более того, как записать обратно в тот же CSV (в соответствующем пустом столбце)

[]

Answer 1

Используйте модуль pandas для импорта csv как объекта DataFrame.Затем вы можете сделать логические операторы для заполнения пустых столбцов:

import pandas as pd
df = pd.read_csv('path_to_file.csv')
df.loc[(df['age']==65) & (df['SES']=='high'), 'vitality_class'] = 1
df.to_csv('path_to_new_file.csv', index=False)