如何找到 Pandas DataFrame 的每一行中的最大值并将相应的列标签添加为新列？

查找每一行中的最大值并使用相应的列标签创建一个新列

您可以使用 pandas 创建一个 DataFrame 并找到每一行的最大值。为此，请使用 idxmax() 函数以及参数 axis=1，该参数指定应在每行的列之间找到最大值。结果将是一个包含最大值的列标签的系列。

要使用这些标签创建新列，您可以将结果分配给 DataFrame 中的列。例如：

import pandas as pd

# Create a DataFrame
df = pd.DataFrame({'Communications and Search': [0.745763, 0.333333, 0.617021, 0.435897, 0.358974],
                   'Business': [0.050847, 0.000000, 0.042553, 0.000000, 0.076923],
                   'General': [0.118644, 0.583333, 0.297872, 0.410256, 0.410256],
                   'Lifestyle': [0.084746, 0.083333, 0.042553, 0.153846, 0.153846]})

# Find the column label of the maximum value for each row
max_labels = df.idxmax(axis=1)

# Create a new column with the maximum values
df['Max'] = max_labels

# Print the resulting DataFrame
print(df)

输出：

   Communications and Search  Business   General  Lifestyle  Max
0                   0.745763  0.050847  0.118644  0.084746  Communications
1                   0.333333  0.000000  0.583333  0.083333     Business
2                   0.617021  0.042553  0.297872  0.042553  Communications
3                   0.435897  0.000000  0.410256  0.153846  Communications
4                   0.358974  0.076923  0.410256  0.153846     Business

这将向 DataFrame 添加一个名为“Max”的新列，其中包含每行最大值的列标签。

以上是如何找到 Pandas DataFrame 的每一行中的最大值并将相应的列标签添加为新列？的详细内容。更多信息请关注PHP中文网其他相关文章！