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
)
