如何在熊猫中添加带有特定列计算的行

时间:2018-08-28 16:55:24

标签: python pandas

我有一个dtaframe,我想在其末尾添加2行,这将指示一个数字范围之间有多少个单元格。除了第一列和最后一列外,我想对所有列都这样做(我有一个包含很多列的大数据框)。 例如,我有以下小型数据框:

    start position      A     b    rnd
0        149449305      4    99     88       
1         47630133      6    50     10   
2        128200594     12     7    600   
3         49423101    100    10      8

我只想计算一个数字在A列和B列的0-9到10-100范围之间的次数,然后将其添加到数据帧的末尾,范围在“开始位置”列的和,就像这样:

        start position      A     b    rnd
    0        149449305      4    99     88       
    1         47630133      6    50     10   
    2        128200594     12     7    600   
    3         49423101    100    10      8
    4              0-9      2     1 
    5           10-100      2     3       

我该怎么做(因为我的数据框很大,而且有很多列,所以没有指明列的名称,而是指明它们的索引范围)?

3 个答案:

答案 0 :(得分:3)

IIUC,将pd.cutvalue_counts一起使用可得到范围计数,然后我们使用append

newdf=df.iloc[:,1:-1].apply(lambda x : pd.cut(x,[0,9,100],labels=['0-9','10-100']).value_counts())
df.append(newdf.rename_axis('startposition',axis=0).reset_index())
Out[216]: 
     A   b    rnd startposition
0    4  99   88.0     149449305
1    6  50   10.0      47630133
2   12   7  600.0     128200594
3  100  10    8.0      49423101
0    2   3    NaN        10-100
1    2   1    NaN           0-9
df=df.append(newdf.rename_axis('startposition',axis=0).reset_index()).reindex(df.columns,axis=1)
df
Out[217]: 
  startposition    A   b    rnd
0     149449305    4  99   88.0
1      47630133    6  50   10.0
2     128200594   12   7  600.0
3      49423101  100  10    8.0
0        10-100    2   3    NaN
1           0-9    2   1    NaN

答案 1 :(得分:1)

您可以将最后两行计算为一个单独的数据框,然后将两个数据框连接起来:

df1 = ...
df2 = pd.DataFrame()
df2['start position'] = ['0-9', '10-99']
df2['A'] = [df1['A'][df1['A'] >= 0][df1['A'] <= 9].count(), df1['A'][df1['A'] >= 10][df1['A'] <= 100].count()]
df2['B'] = [df1['B'][df1['B'] >= 0][df1['B'] <= 9].count(), df1['B'][df1['B'] >= 10][df1['B'] <= 100].count()]

result_df = pd.concat([df1, df2])[['start position','A','B','rnd']].reset_index()

答案 2 :(得分:0)

您只需要在列表中记下一次间隔即可。

intervals=[(0,9), (10,100)]
outside = []
for a,b in intervals:
    inside = []
    for col in df.columns[1:-1]:
        inside.append(df[col].between(a, b).sum())
    outside.append(inside)

new = []
for i, interval in enumerate(intervals):
    new.append(['-'.join(map(str,interval))] + outside[i] + [''])

new_df = pd.concat([df, pd.DataFrame(new, columns=df.columns)], axis=0, ignore_index=True)

  start position    A   b   rnd
0      149449305    4   99  88
1       47630133    6   50  10
2      128200594   12   7   600
3       49423101  100   10  8
4            0-9    2   1   
5         10-100    2   3   

这需要

5.14 ms ± 713 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)