使用 .loc 和 .iloc 进行切片与子集
使用 pandas 进行数据处理
Richie Cotton
Data Evangelist at DataCamp
列表切片
breeds = ["Labrador", "Poodle",
"Chow Chow", "Schnauzer",
"Labrador", "Chihuahua",
"St. Bernard"]
['Labrador',
'Poodle',
'Chow Chow',
'Schnauzer',
'Labrador',
'Chihuahua',
'St. Bernard']
breeds[2:5]
['Chow Chow', 'Schnauzer', 'Labrador']
breeds[:3]
['Labrador', 'Poodle', 'Chow Chow']
breeds[:]
['Labrador','Poodle','Chow Chow','Schnauzer',
'Labrador','Chihuahua','St. Bernard']
切片前先排序索引
dogs_srt = dogs.set_index(["breed", "color"]).sort_index()
print(dogs_srt)
name height_cm weight_kg
breed color
Chihuahua Tan Stella 18 2
Chow Chow Brown Lucy 46 22
Labrador Black Max 59 29
Brown Bella 56 25
Poodle Black Charlie 43 23
Schnauzer Grey Cooper 49 17
St. Bernard White Bernie 77 74
切片外层索引级别
dogs_srt.loc["Chow Chow":"Poodle"]
name height_cm weight_kg
breed color
Chow Chow Brown Lucy 46 22
Labrador Black Max 59 29
Brown Bella 56 25
Poodle Black Charlie 43 23
最后一个值 "Poodle" 被包含
完整数据集
name height_cm weight_kg
breed color
Chihuahua Tan Stella 18 2
Chow Chow Brown Lucy 46 22
Labrador Black Max 59 29
Brown Bella 56 25
Poodle Black Charlie 43 23
Schnauzer Grey Cooper 49 17
St. Bernard White Bernie 77 74
错误切片内层索引级别
dogs_srt.loc["Tan":"Grey"]
Empty DataFrame
Columns: [name, height_cm, weight_kg]
Index: []
完整数据集
name height_cm weight_kg
breed color
Chihuahua Tan Stella 18 2
Chow Chow Brown Lucy 46 22
Labrador Black Max 59 29
Brown Bella 56 25
Poodle Black Charlie 43 23
Schnauzer Grey Cooper 49 17
St. Bernard White Bernie 77 74
正确切片内层索引级别
dogs_srt.loc[
("Labrador", "Brown"):("Schnauzer", "Grey")]
name height_cm weight_kg
breed color
Labrador Brown Bella 56 25
Poodle Black Charlie 43 23
Schnauzer Grey Cooper 49 17
完整数据集
name height_cm weight_kg
breed color
Chihuahua Tan Stella 18 2
Chow Chow Brown Lucy 46 22
Labrador Black Max 59 29
Brown Bella 56 25
Poodle Black Charlie 43 23
Schnauzer Grey Cooper 49 17
St. Bernard White Bernie 77 74
切片列
dogs_srt.loc[:, "name":"height_cm"]
name height_cm
breed color
Chihuahua Tan Stella 18
Chow Chow Brown Lucy 46
Labrador Black Max 59
Brown Bella 56
Poodle Black Charlie 43
Schnauzer Grey Cooper 49
St. Bernard White Bernie 77
完整数据集
name height_cm weight_kg
breed color
Chihuahua Tan Stella 18 2
Chow Chow Brown Lucy 46 22
Labrador Black Max 59 29
Brown Bella 56 25
Poodle Black Charlie 43 23
Schnauzer Grey Cooper 49 17
St. Bernard White Bernie 77 74
双重切片
dogs_srt.loc[
("Labrador", "Brown"):("Schnauzer", "Grey"),
"name":"height_cm"]
name height_cm
breed color
Labrador Brown Bella 56
Poodle Black Charlie 43
Schanuzer Grey Cooper 49
完整数据集
name height_cm weight_kg
breed color
Chihuahua Tan Stella 18 2
Chow Chow Brown Lucy 46 22
Labrador Black Max 59 29
Brown Bella 56 25
Poodle Black Charlie 43 23
Schnauzer Grey Cooper 49 17
St. Bernard White Bernie 77 74
狗狗的日期
dogs = dogs.set_index("date_of_birth").sort_index()
print(dogs)
name breed color height_cm weight_kg
date_of_birth
2011-12-11 Cooper Schanuzer Grey 49 17
2013-07-01 Bella Labrador Brown 56 25
2014-08-25 Lucy Chow Chow Brown 46 22
2015-04-20 Stella Chihuahua Tan 18 2
2016-09-16 Charlie Poodle Black 43 23
2017-01-20 Max Labrador Black 59 29
2018-02-27 Bernie St. Bernard White 77 74
按日期切片
# 获取出生日期在 2014-08-25 到 2016-09-16 之间的狗
dogs.loc["2014-08-25":"2016-09-16"]
name breed color height_cm weight_kg
date_of_birth
2014-08-25 Lucy Chow Chow Brown 46 22
2015-04-20 Stella Chihuahua Tan 18 2
2016-09-16 Charlie Poodle Black 43 23
按部分日期切片
# 获取出生日期在 2014-01-01 到 2016-12-31 之间的狗
dogs.loc["2014":"2016"]
name breed color height_cm weight_kg
date_of_birth
2014-08-25 Lucy Chow Chow Brown 46 22
2015-04-20 Stella Chihuahua Tan 18 2
2016-09-16 Charlie Poodle Black 43 23
按行/列号取子集
print(dogs.iloc[2:5, 1:4])
breed color height_cm
2 Chow Chow Brown 46
3 Schnauzer Grey 49
4 Labrador Black 59
完整数据集
name breed color height_cm weight_kg
0 Bella Labrador Brown 56 25
1 Charlie Poodle Black 43 23
2 Lucy Chow Chow Brown 46 22
3 Cooper Schnauzer Grey 49 17
4 Max Labrador Black 59 29
5 Stella Chihuahua Tan 18 2
6 Bernie St. Bernard White 77 74
Passons à la pratique !
使用 pandas 进行数据处理
