如何使用其他列数据的条件自动填充一列中的行？

Question

我有一个数据框A：

   State     Region   Code
0  Texas     Texas      1
1            Houston    0
2            Dallas     0
3            Austin     0
4  Michigan  Michigan   1
5            Ann Arbor  0
6            Yipsilanti 0
7  Alaska    Alaska     1
8            Troy       0

如果代码=0，我想用上面的状态填写所有状态，期望得到如下输出：

   State     Region   Code   Group
0  Texas     Texas      1      1
1  Texas     Houston    0      1
2  Texas     Dallas     0      1
3  Texas     Austin     0      1
4  Michigan  Michigan   1      2
5  Michigan  Ann Arbor  0      2
6  Michigan  Yipsilanti 0      2
7  Alaska    Alaska     1      3
8  Alaska    Troy       0      3

我尝试添加一个新列"Group"将以上数据分成 3 组，然后使用 groupby 来填写状态

import pandas as pd
import numpy as np
t1 = pd.Series({'State':'Texas', 'RegionalName':'Texas', 'Code':1})
t2 = pd.Series({'State':'  ', 'RegionalName':'Houston','Code' :0})
df=pd.DataFrame([t1,t2])
df.columns=['State','Region','Code']

从 txt.file 读取以生成以上数据帧：

df['Group'] = np.where(df['Code'] == 1, df['Code'+1, df['Code'])

那就不行了。有什么建议吗？谢谢。

Answer 1

你想要 cumsum 代表 Group 和 ffill 代表 State:

df['Group'] = df['Code'].eq(1).cumsum()
df['State'] = df['State'].ffill()

输出：

      State      Region Code  Group
0     Texas       Texas    1      1
1     Texas     Houston    0      1
2     Texas      Dallas    0      1
3     Texas      Austin    0      1
4  Michigan    Michigan    1      2
5       Ann       Arbor    0      2
6       Ann  Yipsilanti    0      2
7    Alaska      Alaska    1      3
8    Alaska        Troy    0      3

如何使用其他列数据的条件自动填充一列中的行？

How to autofill rows in one column with conditioning of other column data?

autofill

pandas