pandasでpaiza問題(配列活用:傾斜配点)

スポンサーリンク

問題へのリンク

listの中に値を入れて、合計値をpandasのsumで計算し、maxで最大値を取得する方法。

import pandas as pd

n = int(input())
weight = [int(_) for _ in input().split()]
lst1 = []

for i in range(n):
    tmp1 = [int(_) for _ in input().split()]
    tmp2 = []
    for j in range(5):
        tmp2.append(tmp1[j] * weight[j])

    lst1.append(tmp2)

df = pd.DataFrame(lst1)
df['sum'] = df.sum(axis=1)
max_value = df['sum'].max()

print(max_value)

コメント

当ブログのコンテンツが気に入ったら広告ブロックの解除(ホワイトリスト化)をご検討下さい。

Please disable your adblocker or whitelist this site!

タイトルとURLをコピーしました