如何在pandas数据集中找到最大值

Question

I have a dataset that shows the share of every disease from total diseases. I want to find the country name which in that country AIDS is bigger than other diseases.

Answer 1

试试

df.index[df.AIDS.eq(df.drop('Total',1).max(1))]

Answer 2

Have a look at pandas max function ( https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.max.html )

Here you could get the max for each row, as is:

most_frequent_disease = df.drop(columns=['Total', 'Other']).max(axis=1)

Then you can create a condition to check wether AIDS is the most frequent disease, and apply it to your dataframe:

is_aids_most_frequent_disease = df.loc[:, 'A'].eq(most_frequent_disease)
df[is_aids_most_frequent_disease]

You could get the country name by using the .index at the end of the expression too.