データフレームのif文の使い方 （複数行の条件）

Question

下記のようなデータフレームがあります．timeは時間:分:秒です．

	time	character
1	00:02:56	Cobb
2	00:03:22	Saito
3	00:05:12	Saito
4	00:05:50	Mal
5	00:06:03	Arthur
6	00:06:37	Mal
7	00:08:21	Cobb
8	00:08:41	Mal

time列の値から60秒以内のとき，character列の内容を結合して表示させるループを作成したいと考えております．

例えば，time列1行目の00:02:56から2行目の00:03:22は60秒以内なのでCobb-Saitoと表示させ，2行目の00:03:22から3行目の00:05:12は60秒以上なので表示させないといったものです．
また，3行目の00:05:12から5行目の00:06:03も同様に60秒以内なので表示させたいので，前後のラインでの比較ではないと考えております．

[理想の結果]
Cobb-Saito #1行目と2行目
Saito-Mal #3行目と4行目
Saito-Arthur #3行目と5行目
Mal-Arthur #4行目と5行目
Mal-Mal #4行目と6行目
Arthur-Mal #5行目と6行目
Cobb-Mal #7行目と8行目

ご教授いただけますでしょうか？
何卒よろしくお願いいたします。

Comments

[meg_]

> また，3行目の00:05:12から5行目の00:06:03も同様に60秒以内なので表示させたいので，前後のラインでの比較ではないと考えております．

ではどういうロジックですか？

[8524ba23]

現状までのコード、あるいはどのような処理を行えばよいのかという基本的な考え、流れを記載ください。
そのうえで、どこ（何）が分からないのかを記載ください。

[ppaul]

timeは文字列ですか、それともdatetime.timeですか？

Answer 1

timeの形式が、文字列なのか、datetime.timeなのかがわからないので、datetime.timeだと仮定しました。
文字列の場合はもう少し簡単になります。
datetime.timeは加減算ができないのでdatetime.timedeltaに変更しています。

>>> print(df)
       time character
1  00:02:56      Cobb
2  00:03:22     Saito
3  00:05:12     Saito
4  00:05:50       Mal
5  00:06:03    Arthur
6  00:06:37       Mal
7  00:08:21      Cobb
8  00:08:41       Mal
>>>
>>> import datetime
>>>
>>> now = datetime.datetime.now()
>>> today = datetime.datetime.combine(now, datetime.time(0))
>>> df['time'] = df['time'].apply(lambda x: datetime.datetime.combine(now, x) - today)
>>>
>>> def near(df, i, distance = datetime.timedelta(seconds=60)):
...     df_result = df[df['time']-df.iloc[i]['time']<=distance].iloc[i+1:]
...     df_result['base_time'] = df.iloc[i]['time']
...     df_result['base_character'] = df.iloc[i]['character']
...     return df_result
...
>>> df_near = pd.concat([near(df, i) for i in range(len(df))])
>>>
>>> df_near['near_character'] = df_near['base_character'] + '-' + df_near['character']
>>>
>>> print(df_near)
             time character       base_time base_character near_character
2 0 days 00:03:22     Saito 0 days 00:02:56           Cobb     Cobb-Saito
4 0 days 00:05:50       Mal 0 days 00:05:12          Saito      Saito-Mal
5 0 days 00:06:03    Arthur 0 days 00:05:12          Saito   Saito-Arthur
5 0 days 00:06:03    Arthur 0 days 00:05:50            Mal     Mal-Arthur
6 0 days 00:06:37       Mal 0 days 00:05:50            Mal        Mal-Mal
6 0 days 00:06:37       Mal 0 days 00:06:03         Arthur     Arthur-Mal
8 0 days 00:08:41       Mal 0 days 00:08:21           Cobb       Cobb-Mal

必要に応じて df_near['near_character'] を取り出して使ってください。

Answer 2

60秒以上なので表示させない
60秒以内なので表示させたい

60秒がどちらかわかりませんね。60秒未満は表示させるにします。

import pandas as pd
import datetime

df = pd.read_csv('data.csv')
print(df)
"""
       time character
0  00:02:56      Cobb
1  00:03:22     Saito
2  00:05:12     Saito
3  00:05:50       Mal
4  00:06:03    Arthur
5  00:06:37       Mal
6  00:08:21      Cobb
7  00:08:41       Mal
"""

df['time'] = pd.to_datetime(df['time'])
df.set_index('time', inplace=True)
ans = list()
td = datetime.timedelta(seconds=60)
for i in range(len(df.index)-1):
    start = str(df.index[i]).split()[1]
    end = str(df.index[i]+td).split()[1]
    lst = df.between_time(start, end, include_end=False)['character'].values.tolist()
    ans.append([lst[0]+'-'+name  for i, name in enumerate(lst) if i > 0])

ans = sum(ans, [])
print(*ans, sep='\n')
"""
Cobb-Saito
Saito-Mal
Saito-Arthur
Mal-Arthur
Mal-Mal
Arthur-Mal
Cobb-Mal
"""