我有以下数据:
data = {
'A': ['car', 'car', 'car', 'car', 'truck', 'truck', 'truck'],
'B': [2, 2, 2, 2, 1, 1, 1]
}
df = pd.DataFrame(data)
字符串
我需要按列A对它进行分组,并将其拆分为包含列B中指定的行数的块。例如,对于列B中的值2,每2行取一行并将其分配给同一组。预期结果是:
outcome = {
'A': ['car', 'car', 'car', 'car', 'truck', 'truck', 'truck'],
'B': [2, 2, 2, 2, 1, 1, 1],
'C': [0, 0, 1, 1, 0, 1, 2]
}
型
其中列C是组编号。
1条答案
按热度按时间imzjd6km1#
您可以尝试:
字符串
注:假设对于每个A,B是相同的。