python - 向不退出的行添加值

Question

假设我有一个 pandas 数据框，如下所示:

row_number     value
1              21
3              10234
5              467
6              395
7              762

# notice row_number == 2 and == 4 are missing

对于row_number < 10(例如)的每个尚未存在行，我想插入value == 0的行.也就是说，我想获得以下数据集:

row_number     value
1              21
3              10234
5              467
6              395
7              762
2              0
4              0
8              0
9              0

我什至无法“选择”这些不存在的行。我试过这个:

my_df[my_df["row_number"] == i for i in range(10) if i not in 
my_df["row_number"]]

但是我收到无效语法错误。

怎样才能实现我想要的目标？谢谢。

Answer 1

使用reindex按范围 set_index和 reset_index 索引:

df = df.set_index('row_number').reindex(range(1,10), fill_value=0).reset_index()
print (df)
   row_number  value
0           1     21
1           2      0
2           3  10234
3           4      0
4           5    467
5           6    395
6           7    762
7           8      0
8           9      0

如果顺序很重要:

a = df['row_number'].values.tolist() + np.setdiff1d(np.arange(1,10), df['row_number'].values).tolist()
print (a)
[1, 3, 5, 6, 7, 2, 4, 8, 9]

df = df.set_index('row_number').reindex(a, fill_value=0).reset_index()
print (df)
   row_number  value
0           1     21
1           3  10234
2           5    467
3           6    395
4           7    762
5           2      0
6           4      0
7           8      0
8           9      0