pandasでpaiza問題(配列活用:二人三脚)

スポンサーリンク

paizaへのリンク(配列活用:二人三脚 (paizaランク C 相当)

pandasのdiffを使って差分を取得し、idxminで最小値の位置を特定する練習。idxminで返される値のtypeはint。

import pandas as pd

n = int(input())
lst = []

for i in range(n):
    lst.append(int(input()))

df = pd.DataFrame(lst)

df = df.sort_values(0)
df = df.reset_index(drop=True)
df['diff'] = df.diff()

min_idx = df['diff'].idxmin()

print(df.iat[min_idx - 1, 0])
print(df.iat[min_idx, 0])

コメント

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

Please disable your adblocker or whitelist this site!

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