Найдите максимальное значение из нескольких столбцов и верните его соответствующее имя столбца для каждого состояния ⇐ Python

[Anonymous]

Я пытаюсь найти самый высокий год с наибольшим количеством рождений для каждого штата (есть 5 различных столбцов на протяжении многих лет, 2016, 2017 г. и т. Д.), И мне нужно сделать новый столбец для каждого штата, чтобы показать название колонка с самыми высокими родами. Как мне это сделать?data = { 'State': ['California', 'Texas', 'New York'], '2018': [53000, 42000, 28000], '2019': [51000, 43000, 27000], '2020': [50000, 44000, 26000] }
df = pd.DataFrame(data) # year with the highest births for each state
df['Highest_Birth_Year'] = df.iloc[:, 1:].idxmax(axis=1)
print(df)
< /code>
Выше я ищу. Я знаю, как перечислить штаты, но я не могу перечислить каждый номер по каждому году, так как их слишком много, как мне это сделать? Спасибо

Подробнее здесь: https://stackoverflow.com/questions/794 ... ing-column