簡體   English   中英

Pandas-創建一個基於列值插入新行的表?

[英]Pandas- creating a table that inserts new rows based on column values?

我有一個數據框,其中包含項目的名稱,數據,然后是競爭對手的數據:

 name   value1   value2    ex_value1     ex_value2   
 jim       0.4      0.6           0.7           0.3  
 tim       0.2      0.8   0.766666667   0.233333333  
 john        1        0           0.5           0.5  
 paul      0.9      0.1   0.533333333   0.466666667  

我想要做的是創建一個按名稱索引的新表,但根據競爭對手的數據插入新行,以便顯示jim,ex-jim,tim,ex-tim等:

   name       value1        value2     
  jim               0.4           0.6  
  tim               0.2           0.8  
  john                1             0  
  paul              0.9           0.1  
  ex_jim            0.7           0.3  
  ex_tim    0.766666667   0.233333333  
  ex_john           0.5           0.5  
  ex_paul   0.533333333   0.466666667  

我該怎么做呢? 我是否必須在名稱上設置索引,然后以這種方式插入新的? 我會通過一個循環得到這個嗎? 對此表示贊賞

你可以使用concat來做到這一點

df_ex = df[['name','ex_value1', 'ex_value2']].rename(columns = {'ex_value1': 'value1', 'ex_value2': 'value2'})

df_ex['name'] = 'ex_' + df_ex['name']

pd.concat([df[['name','value1', 'value2']], df_ex ]).round(2)

    name    value1  value2
0   jim     0.40    0.60
1   tim     0.20    0.80
2   john    1.00    0.00
3   paul    0.90    0.10
0   ex_jim  0.70    0.30
1   ex_tim  0.77    0.23
2   ex_john 0.50    0.50
3   ex_paul 0.53    0.47

我想重新創建df,你可以在最后添加reset_index()

pd.DataFrame(df.iloc[:,1:].values.reshape(8,2),index=['','ex_']*4+df.name.repeat(2),columns=['value1','value2'])
Out[986]: 
           value1    value2
name                       
jim      0.400000  0.600000
ex_jim   0.700000  0.300000
tim      0.200000  0.800000
ex_tim   0.766667  0.233333
john     1.000000  0.000000
ex_john  0.500000  0.500000
paul     0.900000  0.100000
ex_paul  0.533333  0.466667

我建議將您的數據幀拆分為兩個,然后再將它們重新連接起來。 就像是:

import pandas as pd

df = pd.DataFrame([['jim', .4, .6, .7, .3], ['john', 1, 0, .5, .5]], columns=['name', 'value1', 'value2', 'ex_value1', 'ex_value2'])

ex_df = df.copy()

ex_df['name'] = 'ex_'+ex_df['name'].astype(str)

ex_df = ex_df[['name', 'ex_value1', 'ex_value2']]
ex_df.columns = ['name', 'value1', 'value2']

df = df[['name', 'value1', 'value2']]

frames = (df, ex_df)

new = pd.concat(frames).reset_index()
new = new[['name', 'value1', 'value2']]

print(new)

#output
         name  value1  value2
0      jim     0.4     0.6
1     john     1.0     0.0
2   ex_jim     0.7     0.3
3  ex_john     0.5     0.5

你可以去

def myfunc(row):
    return pd.Series({'name': 'ex_{}'.format(row['name']), 
                      'value1': row['ex_value1'], 
                      'value2': row['ex_value2']})

df2 = df[~df['name'].astype(str).str.startswith('ex_')].apply(myfunc,axis =1)
df = pd.concat([df[['name', 'value1', 'value2']], df2])

這僅將函數myfunc應用於name不以ex_開頭的行。 myfunc()返回一個新的數據幀,然后連接到df


對於單線愛好者(雖然不可取,但確實如此):

 df = pd.concat([df[['name', 'value1', 'value2']], df[~df['name'].astype(str).str.startswith('ex_')].apply(myfunc,axis = 1)]) 

你可以使用meltpivot的組合

df2 = df.melt('name')
df2.loc[df2.variable.str.contains('ex'),'name'] = 'ex_' +df2.name
df2.variable = df2.variable.str.strip('ex_')
df2 = df2.pivot(index='name',columns='variable').reset_index()
df2.columns = df2.columns.droplevel(0)

給你的

variable             value1    value2
0          ex_jim  0.700000  0.300000
1         ex_john  0.500000  0.500000
2         ex_paul  0.533333  0.466667
3          ex_tim  0.766667  0.233333
4             jim  0.400000  0.600000
5            john  1.000000  0.000000
6            paul  0.900000  0.100000
7             tim  0.200000  0.800000

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM