Pythonで各行の最大値を取得する方法

Pythonでデータ分析を行う際、各行や各列の最大値を取得することはよくあります。ここでは、PythonのライブラリであるPandasとNumpyを使用して、各行の最大値を取得する方法を紹介します。

Pandasを使用した方法

PandasのDataFrameでは、max関数を使用して各行または各列の最大値を取得することができます。以下にその使用例を示します。

import pandas as pd

# DataFrameを作成
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
})

# 各行の最大値を取得
df['max'] = df.max(axis=1)

print(df)

このコードでは、新たな列maxが作成され、その列に各行の最大値が格納されます。

Numpyを使用した方法

Numpyでは、amax関数を使用して各行または各列の最大値を取得することができます。以下にその使用例を示します。

import numpy as np

# 配列を作成
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 各行の最大値を取得
x = np.amax(a, axis=1)

print(x)

このコードでは、各行の最大値が格納された配列xが出力されます。

以上、Pythonで各行の最大値を取得する方法を紹介しました。データ分析を行う際には、これらの方法を活用してみてください。