2

我在 python pandas 中有两个系列。

一个来自名为values.csv. 它看起来像这样：

time, value
0, 10312435
9, 45924523
11, 43423434
20, 42343552
...

另一个被调用breaks.csv，它看起来像这样：

time
5
18
...

问题：我想values.csv根据breaks.csv.

在上面的示例中，第一个断点是5，导致文件或集合包含其中的所有条目time \in [0, 5]，因此只有值0, 10312435。第二个断点是18，因此第二批值应该在(5, 18]ie之内9, 45924523，11, 43423434以此类推。

在 pandas（或者其他一些易于使用的 python 包）中是否有可能发生这样的事情？

4

2 回答 2

2

您可以先从这些垃圾箱中形成垃圾箱breaks.time，然后values.time使用这些垃圾箱分配类别pd.cut：

import numpy as np

# intervals to fall into
bins = [-np.inf, *breaks.time, +np.inf]

# distinct labels of 0..N-1
labels = np.arange(len(bins) - 1)

# form a new column in `values` with assigned categories
values["cats"] = pd.cut(values.time, bins=bins, labels=labels)

此时values看起来像：

>>> values

   time     value cats
0     0  10312435    0
1     9  45924523    1
2    11  43423434    1
3    20  42343552    2

现在我们可以分组cats，例如，形成一个数据框列表：

# no need for `cats` column anymore, so we drop it when putting in
frames_list = [frame.drop(columns="cats")
               for _, frame in values.groupby("cats")[["time", "value"]]]

我们可以访问框架

>>> frames_list[0]

   time     value
0     0  10312435


>>> frames_list[1]

   time     value
1     9  45924523
2    11  43423434

>>> frames_list[2]

   time     value
3    20  42343552
于 2021-06-28T07:48:10.863 回答
0

我想出了以下内容，基于Pandas 按列值拆分 DataFrame

sim_dist_right = pandas.read_csv('sim/dist_right.csv', comment='#')
sim_round_indicator = pandas.read_csv('sim/round_indicator.csv', comment='#')

round_list = []
for index, row in sim_round_indicator.iterrows():
    print("splitting at " + str(row['time']))
    df_sep = sim_dist_right[sim_dist_right['time'] < row['time']]
    
    round_list.append(df_sep)
    print("separated a batch of " + str(len(df_sep)) + " elements")
    
    df_over = sim_dist_right[sim_dist_right['time'] >= row['time']]
    print(str(len(df_over)) + " elements over")
    
    sim_dist_right = df_over
    
print("splitted values into " + str(len(round_list)) + " batches")
于 2021-06-28T07:48:57.673 回答