分享8|分享8 个常用pandas的 index设置
目录
- 1. 将索引从 groupby 操作转换为列
- 2. 使用现有的 DataFrame 设置索引
- 3. 一些操作后重置索引
- 4.排序后重置索引
- 5.删除重复后重置索引
- 6. 索引的直接赋值
- 7.写入CSV文件时忽略索引
- 8.读取时指定索引列
文章图片
1. 将索引从 groupby 操作转换为列
groupby
分组方法是经常用的。比如下面通过添加一个分组列team来进行分组。>>> df0["team"] = ["X", "X", "Y", "Y", "Y"]>>> df0ABC team00.5480120.2885830.734276X10.3428950.2079170.995485X20.3787940.1609130.971951Y30.0397380.0084140.226510Y40.5810930.7503310.133022Y>>> df0.groupby("team").mean()ABCteamX0.4454530.2482500.864881Y0.3332080.3065530.443828
默认情况下,分组会将分组列编程
index
索引。但是很多情况下,我们不希望分组列变成索引,因为可能有些计算或者判断逻辑还是需要用到该列的。因此,我们需要设置一下让分组列不成为索引,同时也能完成分组的功能。有两种方法可以完成所需的操作,第一种是用
reset_index
,第二种是在groupby
方法里设置as_index=False
。个人更喜欢第二种方法,它只涉及两个步骤,更简洁。>>> df0.groupby("team").mean().reset_index()teamABC0X0.4454530.2482500.8648811Y0.3332080.3065530.443828>>> df0.groupby("team", as_index=False).mean()teamABC0X0.4454530.2482500.8648811Y0.3332080.3065530.443828
2. 使用现有的 DataFrame 设置索引
当然,如果已经读取数据或做完一些数据处理步骤后,我们可以通过
set_index
手动设置索引。>>> df = pd.read_csv("data.csv", parse_dates=["date"])>>> df.set_index("date")temperaturehumiditydate2021-07-0195502021-07-0294552021-07-039456
这里有两点需要注意下:
set_index
方法默认将创建一个新的DataFrame
。如果要就地更改df的索引,需要设置inplace=True
。
df.set_index(“date”, inplace=True)
- 如果要保留将要被设置为索引的列,可以设置
drop=False
。
df.set_index(“date”, drop=False)
3. 一些操作后重置索引
在处理
DataFrame
时,某些操作(例如删除行、索引选择等)将会生成原始索引的子集,这样默认的数字索引排序就乱了。如要重新生成连续索引,可以使用reset_index
方法。>>> df0 = pd.DataFrame(np.random.rand(5, 3), columns=list("ABC"))>>> df0ABC00.5480120.2885830.73427610.3428950.2079170.99548520.3787940.1609130.97195130.0397380.0084140.22651040.5810930.7503310.133022>>> df1 = df0[df0.index % 2 == 0]>>> df1ABC00.5480120.2885830.73427620.3787940.1609130.97195140.5810930.7503310.133022>>> df1.reset_index(drop=True)ABC00.5480120.2885830.73427610.3787940.1609130.97195120.5810930.7503310.133022
通常,我们是不需要保留旧索引的,因此可将drop参数设置为True。同样,如果要就地重置索引,可设置
inplace
参数为True,否则将创建一个新的 DataFrame
。4.排序后重置索引
当用
sort_value
排序方法时也会遇到这个问题,因为默认情况下,索引index
跟着排序顺序而变动,所以是乱雪。如果我们希望索引不跟着排序变动,同样需要在sort_values
方法中设置一下参数ignore_index
即可。>>> df0.sort_values("A")ABC team30.0397380.0084140.226510Y10.3428950.2079170.995485X20.3787940.1609130.971951Y00.5480120.2885830.734276X40.5810930.7503310.133022Y>>> df0.sort_values("A", ignore_index=True)ABC team00.0397380.0084140.226510Y10.3428950.2079170.995485X20.3787940.1609130.971951Y30.5480120.2885830.734276X40.5810930.7503310.133022Y
5.删除重复后重置索引
删除重复项和排序一样,默认执行后也会打乱排序顺序。同理,可以在
drop_duplicates
方法中设置ignore_index
参数True
即可。>>> df0ABC team00.5480120.2885830.734276X10.3428950.2079170.995485X20.3787940.1609130.971951Y30.0397380.0084140.226510Y40.5810930.7503310.133022Y>>> df0.drop_duplicates("team", ignore_index=True)ABC team00.5480120.2885830.734276X10.3787940.1609130.971951Y
6. 索引的直接赋值
当我们有了一个
DataFrame
时,想要使用不同的数据源或单独的操作来分配索引。在这种情况下,可以直接将索引分配给现有的 df.index
。>>> better_index = ["X1", "X2", "Y1", "Y2", "Y3"]>>> df0.index = better_index>>> df0ABC teamX10.5480120.2885830.734276XX20.3428950.2079170.995485XY10.3787940.1609130.971951YY20.0397380.0084140.226510YY30.5810930.7503310.133022Y
7.写入CSV文件时忽略索引
数据导出到 CSV 文件时,默认
DataFrame
具有从 0 开始的索引。如果我们不想在导出的 CSV 文件中包含它,可以在to_csv
方法中设置index
参数。>>> df0.to_csv("exported_file.csv", index=False)
如下所示,导出的 CSV 文件中,索引列未包含在文件中。
文章图片
【分享8|分享8 个常用pandas的 index设置】其实,很多方法中都有关于索引的设置,只不过大家一般比较关心数据,而经常忽略了索引,才导致继续运行时可能会报错。以上几个高频的操作都是有索引设置的,建议大家平时用的时候养成设置索引的习惯,这样会节省不少时间。
8.读取时指定索引列
很多情况下,我们的数据源是 CSV 文件。假设有一个名为的文件
data.csv
,包含以下数据。date,temperature,humidity07/01/21,95,5007/02/21,94,5507/03/21,94,56
默认情况下,pandas将会创建一个从0开始的索引行,如下:
>>> pd.read_csv("data.csv", parse_dates=["date"])datetemperaturehumidity0 2021-07-0195501 2021-07-0294552 2021-07-039456
但是,我们可以在导入过程中通过将
index_col
参数设置为某一列可以直接指定索引列。>>> pd.read_csv("data.csv", parse_dates=["date"], index_col="date")temperaturehumiditydate2021-07-0195502021-07-0294552021-07-039456
到此这篇关于分享8 个常用pandas的 index设置的文章就介绍到这了,更多相关常用pandas的 index设置内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
推荐阅读
- 放屁有这三个特征的,请注意啦!这说明你的身体毒素太多
- 一个人的旅行,三亚
- 一个小故事,我的思考。
- 一个人的碎碎念
- 第326天
- 猎杀IP
- 七年之痒之后
- 我从来不做坏事
- 喂,你结婚我给你随了个红包
- 异地恋中,逐渐适应一个人到底意味着什么()