Python: how to do basic data manipulation like in R?

前端 未结 4 1149
半阙折子戏
半阙折子戏 2021-01-31 21:22

I have been working with R for several years. R is very strong in data manipulation. I\'m learning python and I would like to know how to manipulate data using python. Basically

4条回答
  •  滥情空心
    2021-01-31 21:32

    Simple answer: use pandas

    1

    In [2]: df = read_csv('foo.csv', index_col=None)
    
    In [3]: df
    Out[3]: 
       var1  var2  var3
    0  1     2     3   
    1  4     5     6   
    2  7     8     9   
    

    2

    In [4]: df[df['var2'].isin([5, 8])]
    Out[4]: 
       var1  var2  var3
    1  4     5     6   
    2  7     8     9   
    

    3

    In [5]: df['var4'] = df['var3'] * 2
    
    In [6]: df
    Out[6]: 
       var1  var2  var3  var4
    0  1     2     3     6   
    1  4     5     6     12  
    2  7     8     9     18  
    

    4

    In [7]: df.T
    Out[7]: 
          0  1   2 
    var1  1  4   7 
    var2  2  5   8 
    var3  3  6   9 
    var4  6  12  18
    

    6

    In [8]: df.to_csv('foo2.csv')
    
    In [9]: !cat foo2.csv
    index,var1,var2,var3,var4
    0,1,2,3,6
    1,4,5,6,12
    2,7,8,9,18
    

提交回复
热议问题