Python 常用函数方法、技巧

Python 常用函数方法

DataFrame 与 Dict 的转换、reindex()对DataFrame排序

注：字典没有顺序，DataFrame有顺序。

import pandas as pd
import numpy as np
dict = {'c':1,'a':2,'b':3}
print(dict) #打印字典
#将字典转为 DataFrame,列索引命名为zhi,行索引为原字典的键名
df = pd.DataFrame([dict],index=['zhi'])  #DataFrame转Dict
#按列表对DataFrame进行排序
df_sort = df.T.reindex(['b','c','a'])
print(df_sort)
#reset_index：原行索引作为一列保留,列名改为 ID
df_setindex = df_sort.reset_index().rename(columns={'index':'ID'})
print(df_setindex)
#将DataFrame转为字典,取ID为字典的键,zhi为字典的值
df_dict = df_setindex.set_index('ID').T.to_dict('list') #Dict转DataFrame
print(df_dict) #打印字典
------------------------------------------------------
字典==> {'c': 1, 'b': 3, 'a': 2}
   zhi
b    3
c    1
a    2
  ID  zhi
0  b    3
1  c    1
2  a    2
字典==> {'c': [1], 'b': [3], 'a': [2]}

Python DataFrame.groupby()聚合函数

groupby()返回值格式为DataFrame，特别注意，groupby里面的字段内的数据重构后都会变成索引；groupby(),一般和sum()、mean()一起使用。

import pandas as pd
df = pd.read_excel('temp_python.xlsx')
#print(df)
df_grp = df.groupby(['证券代码', '买卖方向']).sum()
df_grp.to_csv("myexcel2.csv")   # 导出数据到CSV文件myexcel2.csv
print(df_grp)

import pandas as pd
df = pd.read_excel('temp_python.xlsx')
#print(df)
df.index = [df['证券代码'], df['买卖方向']]
df.to_csv("myexcel2.csv")   # 导出数据到CSV文件myexcel2.csv
print(df)