python - 如何根据多列的值拆分数据框
问题描述
我正在使用 Python。我想根据两列的值拆分我的数据框。每次值对发生变化时,我都想在这个位置拆分我的数据框。
例子:
df = pd.DataFrame({'Distance':[1,1,1,1,3,3,3], 'labels':[1,2,2,2,4,4,5]})
df=
Distance labels
0 1 1
1 1 2
2 1 2
3 1 2
4 3 4
5 3 4
6 3 5
我想得到:
list_of_dfs[0]=
Distance labels
0 1 1
list_of_dfs[1]=
Distance labels
1 1 2
2 1 2
3 1 2
list_of_dfs[2]=
Distance labels
4 3 4
5 3 4
list_of_dfs[3]=
Distance labels
6 3 5
这是它的工作原理:
l = [1,4,6,7]
l_mod = [0] + l + [max(l)+1]
list_of_dfs = [df.iloc[l_mod[n]:l_mod[n+1]] for n in range(len(l_mod)-1)]
我的问题:
如何自动获取数组 l=[1,4,6,7] ?这就是我完成这项任务所需要的一切!
解决方案
使用 pd.duplicates 查找唯一行。
# use duplicated to determine rows that are original and create list
ind = df[~df.duplicated()].index.tolist()
# account for the last row, append value one greater than maximum index.
ind.append(df.shape[0])
# create dictionary for dataframe.
dfs = {}
# use iloc to create new dataframes, then add to dictionary.
for i in range(len(ind)-1):
df_temp = df.iloc[ind[i]:ind[i+1], :]
dfs[i] = df_temp
从字典中检索您的数据框:
df0 = dfs[0]
Distance labels
0 1 1
df1 = dfs[1]
print(df1)
Distance labels
1 1 2
2 1 2
3 1 2
推荐阅读
- firebase - 多租户的 Firebase 身份验证和 Apple 登录
- python - Python Plotly图例按升序显示
- node.js - 使用 Stripe Firebase 扩展 Webhook 运行订阅付款未触发
- apache-kafka - Confluent 控制中心是否可扩展?
- angular - Angular 9 - 如何将动态参数注入服务构造函数
- xml - 仅使用 XML Schema 1.0 验证子树
- powershell - 通过 Start-Process 启动的脚本在 90% 的情况下都有效 - 当它“失败”时,它只是变成了一个僵尸并且不执行第一行代码
- dynamics-crm - 从数组中提取元素,然后使用逻辑应用更新它们的字段
- c# - 一年后,应用程序现在抛出:System.Net.Sockets.SocketException:现有连接被远程主机强行关闭
- html - 仅当浏览器上的屏幕缩小时,如何才能显示消息?