Question

我导入了CSV数据集，但重组数据时遇到了问题。数据如下：

1    2    3    4
UK   NaN  NaN  NaN
a    b    c    d
b    d    c    a
.    .    .    .
US   NaN  NaN  NaN
a    b    c    d
.    .    .    .

我想在英国，美国等地添加一个新列，例如：

area    1    2   3   4
UK      a    b   c   d
UK      b    d   c   a
 .      .    .   .   .
US      a    b   c   d

这需要适用于中间数据不同的多个区域。

提前致谢。

Answer 1

这是单程

In [4461]: nn =  df['2'].notnull()

In [4462]: df[nn].assign(area=df['1'].mask(nn).ffill())
Out[4462]:
   1  2  3  4 area
1  a  b  c  d   UK
2  b  d  c  a   UK
4  a  b  c  d   US

Answer 2

按位置使用insert作为新列：

print (df[1].where(df[2].isnull()).ffill())
0    UK
1    UK
2    UK
3    US
4    US
Name: 1, dtype: object

df.insert(0, 'area', df[1].where(df[2].isnull()).ffill())
#alternative
#df.insert(0, 'area', df[1].mask(df[2].notnull()).ffill())
df = df[df[1] != df['area']].reset_index(drop=True)
print (df)
  area  1  2  3  4
0   UK  a  b  c  d
1   UK  b  d  c  a
2   US  a  b  c  d

在没有第一列的情况下检查所有NaN的另一种解决方案：

print (df[1].where(df.iloc[:, 1:].isnull().all(1)).ffill())
0    UK
1    UK
2    UK
3    US
4    US
Name: 1, dtype: object

Pandas多行名称为列

2 个答案: