繁体   English   中英

将第一行和标题与熊猫结合起来

[英]Combine first row and header with pandas

我正在尝试在 Python 中复制我在 R 中编写的代码,但在尝试修复标头时遇到了困难。 我想合并标题和第一行,但我似乎无法让它工作。

我的数据目前看起来像这样:

        Acronym  Project Number    Title     Dates         Unnamed: 4   Unnamed: 5 
0         NaN            NaN        NaN      Start date    Duration     Number of Participants
1         YoYo           5678       blabla   01-01-2020    36           4
2         SPY            1452       blabla   06-03-2018    12           6         
3         NoNo           6280       blabla   02-05-2019    48           8       
4         MaGiK          2749       blabla   01-05-2016    12           9    

我的数据标题分布在两行中。 前三个标题是他们应该的。 但是,从第 4 列开始,我的列名称改为在第二行。 简单地组合标题和第一行是行不通的,因为某些列(如第 4 列)在标题和第一行中都有值。

理想情况下,我想做的是将标题保留在前三列中,并将第一行中的值用作第 4 列以后的标题,这样它就变成了:

Acronym | Project Number | Title | Start date  | Duration | Number of Participants

在 R 中,这将通过

Projects <- Projects %>% 
  set_names(c(names(Projects)[1:3], as.character(.[1,])[4:6])) %>% 
  rownames_to_column("index") %>% 
  filter(index != 1 ) %>% 
  select(-index)

基于这个问题,我在 Python 中尝试了这个

Projects_clean.columns = np.concatenate([Projects_clean.iloc[0, :2], Projects_clean.columns[0:3]])

Projects_clean.columns = np.append(Projects_clean.iloc[0,:2], Projects_clean.columns[0:3])

但这返回

 "values have {new} elements".format(old=old_len, new=new_len)

Length mismatch: Expected axis has 24 elements, new values have 10 elements

(根据我的实际数据)。 显然,我的新数据没有与旧数据相同数量的列。 我究竟做错了什么?

在第 3 个值之后转换前 3 列名称和第一行的所有数据:

Projects_clean.columns = (Projects_clean.columns[:3].tolist() +  
                          Projects_clean.iloc[0, 3:].tolist())

要么:

Projects_clean.columns = np.concatenate([Projects_clean.columns[:3], 
                                         Projects_clean.iloc[0, 3:]])

然后:

Projects_clean = Projects_clean.iloc[1:]
print (Projects_clean)
  Acronym  Project Number   Title  Start date Duration Number of Participants
1    YoYo          5678.0  blabla  01-01-2020       36                      4
2     SPY          1452.0  blabla  06-03-2018       12                      6
3    NoNo          6280.0  blabla  02-05-2019       48                      8
4   MaGiK          2749.0  blabla  01-05-2016       12                      9

如果可能,通过read_csv中的参数header=[0,1]通过前 2 行创建MultiIndex ,然后使用:

Projects_clean = pd.read_csv(file, header=[0,1])


Projects_clean.columns = (Projects_clean.columns.get_level_values(0)[:3].tolist() +  
                          Projects_clean.columns.get_level_values(1)[3:].tolist())
print (Projects_clean)
  Acronym  Project Number   Title  Start date  Duration  \
1    YoYo            5678  blabla  01-01-2020        36   
2     SPY            1452  blabla  06-03-2018        12   
3    NoNo            6280  blabla  02-05-2019        48   
4   MaGiK            2749  blabla  01-05-2016        12   

   Number of Participants  
1                       4  
2                       6  
3                       8  
4                       9  

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM