如何解决在Python中用相同的数字计算每两行
我有一个每两行相关的数据框。我试图给每两行一个唯一的 ID。我认为这会容易得多,但我无法弄清楚。假设我有这个数据框:
df = pd.DataFrame({'Var1': ['A',2,'C',7],'Var2': ['B',5,'D',9]})
print(df)
Var1 Var2
A B
2 5
C D
7 9
我想添加一个 ID 来生成如下所示的数据框:
df = pd.DataFrame({'ID' : [1,1,2],'Var1': ['A',9]})
print(df)
ID Var1 Var2
1 A B
1 2 5
2 C D
2 7 9
这只是一个示例,但每两行都是相关的,因此只需尝试在 ID 列中按 1、1、2、2、3、3 等计数。
感谢您的帮助!
解决方法
您可以先创建一个序列,然后将其除以 2(整数除法):
import numpy as np
df['ID'] = np.arange(len(df)) // 2 + 1
df
# Var1 Var2 ID
#0 A B 1
#1 2 5 1
#2 C D 2
#3 7 9 2
,
我认为这不是 Pandas 的原生方式,但这是可行的...
import pandas as pd
df = pd.DataFrame({'Var1': ['A',2,'C',7],'Var2': ['B',5,'D',9]})
df['ID'] = 1 + df.index // 2
df[['ID','Var1','Var2']]
输出:
ID Var1 Var2
0 1 A B
1 1 2 5
2 2 C D
3 2 7 9
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。