Python3 二重リスト 共通要素のリスト作成

Question

N行与えられた整数がある

それぞれ３つの整数が空白区切りで与えられる

左から　x y z　とする

x,y　は、 x 時 y 分 を示す

z は　歩いた歩数を示す

与えられたデータから、時間ごとにあるいた歩数の総数を求めたい。

入力例1
5

3 20 70
3 40 170
3 59 90
4 5 55
4 25 40

出力例1
330 95

入力例2

10

1 10 1
1 20 1
1 30 1
1 40 1
1 50 1
2 10 1
2 20 2
2 30 3
2 40 4
2 50 5

出力例2
5 14

解釈)

データの時間が共通している部分ごとに新しいリストを作りたい

data[0][0] == data[1][0]　の条件に一致しているが同じ時間帯になる

以下のコードだとエラーになってしまう。

ヒントなどあれば教えて頂きたいです。

コード

M = int(input())
N = int(input())

data = [list(map(int,input().split())) for _  in range(N)]

for x in range(N):
    if data[x][0] == data[x+1][0]:
        print("OK")


出力
	
Traceback (most recent call last):
  File "Main.py", line 18, in <module>
    if data[x][0] == data[x+1][0]:
IndexError: list index out of range

Answer 1

ヒントだけ。
groupbyのkeyを指定すればできそうです。

10.1. itertools — 効率的なループ実行のためのイテレータ生成関数 — Python 3.6.5 ドキュメント

ついでに

この処理ですが、

for x in range(N):
    if data[x][0] == data[x+1][0]:
        print("OK")

x+1するとリストのインデックスの上限をはみ出してしまうのは当然です。
安直な対処法ですが、range(N-1)にするととりあえず落ちなくはなります。

Comments

[python3_beginer]

便利な機能ですね！
すごく勉強になりました。
ご指導ありがとうございました。

Answer 2

hayataka2049さんヒントのgroupbyを用いた方法です。

from itertools import groupby
data = [
    [3, 20, 70],
    [3, 40, 170],
    [3, 59, 90],
    [4, 5,  55],
    [4, 25, 40]
] # 時でソートされていること。
result = []
for hour, data_walk in groupby(data, key=lambda x: x[0]):
    walk_per_hour = sum(x[2] for x in list(data_walk))
    result.append(walk_per_hour)
print(result)

Comments

[python3_beginer]

便利な機能ですね！
スマートなコードで助かりました。
ご指導ありがとうございました。

Answer 3

dataをpandasのDataFrameに変換して時間をキーに集計するメソッドを使うというのはいかがでしょうか。

# dataを生成するまでは省略
import pandas as pd
df_data=pd.DataFrame(data,columns=['HH','MM','num'])
result=df_data.groupby('HH').sum()['num']

これだと、「時間ごとの歩数の合計」をそのままコードにできます。

Answer 4

素朴に。

from io import StringIO
import sys

s = '''10
1 10 1
1 20 1
1 30 1
1 40 1
1 50 1
2 10 1
2 20 2
2 30 3
2 40 4
2 50 5'''

a = StringIO(s)

def input():
    return next(a)

#------------------------------------------------
 
N = int(input())
data = [[int(v) for v in input().split()] for _  in range(N)]

x, y, z = data[0]
p = x
r = z
ans = []
for x, y, z in data[1:]:
    if x == p:
        r += z
    else:
        ans.append(r)
        p = x
        r = z
ans.append(r)

print(*ans)

Comments

[python3_beginer]

簡潔で分かりやすいですね！
すごく勉強になりました。
ご指導ありがとうございました。