python - 使用 pandas 数据框计算行数

Question

我有一个带有时间戳数据的 pandas 数据框。我想计算每分钟有多少行，例如，给定此表:

00:00:01 数据1

00:00:21 数据2

00:02:02 数据3

00:02:10 数据4

00:02:31 数据5

结果应该是[2,0,3]。我该怎么做？

Answer 1

您可以首先将第一列转换为 pd.Timedelta dtype，然后将其设置到索引中并使用带有计数的重新采样:

from io import StringIO
txt = StringIO("""00:00:01 data1

00:00:21 data2

00:02:02 data3

00:02:10 data4

00:02:31 data5""")

df = pd.read_csv(txt, names=['Time','Value'], sep='\s+')


df['Time'] = pd.to_timedelta(df['Time'])

df = df.set_index('Time')

df_out = df.resample('T').count()

df_out 

输出:

          Value
Time           
00:00:01      2
00:01:01      0
00:02:01      3

列表:

df_out['Value'].tolist()

输出:

[2, 0, 3]