dataframe选取行、列方法总结。ix、iloc、loc等
pandas中的切片方法
[],loc,iloc,at,iat,ix
基础数据
import pandas as pd
import random
random.seed(0)
rnd_1 = [random.randrange(1,20) for x in range(10)]
rnd_2 = [random.randrange(1,20) for x in range(10)]
rnd_3 = [random.randrange(1,20) for x in range(10)]
fecha = pd.date_range('2012-4-10', '2012-4-19')
data = pd.DataFrame({'fecha':fecha, 'rnd_1': rnd_1, 'rnd_2': rnd_2, 'rnd_3': rnd_3})
print(data)
print(data.describe())
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
结果：
fecha rnd_1 rnd_2 rnd_3
0 2012-04-10 13 19 10
1 2012-04-11 14 7 4
2 2012-04-12 2 17 3
3 2012-04-13 9 5 11
4 2012-04-14 17 10 16
5 2012-04-15 16 5 18
6 2012-04-16 13 4 4
7 2012-04-17 10 9 12
8 2012-04-18 16 18 14
9 2012-04-19 12 5 11
rnd_1 rnd_2 rnd_3
count 10.000000 10.000000 10.000000
mean 12.200000 9.900000 10.300000
std 4.417138 5.915141 5.186521
min 2.000000 4.000000 3.000000
25% 10.500000 5.000000 5.500000
50% 13.000000 8.000000 11.000000
75% 15.500000 15.250000 13.500000
max 17.000000 19.000000 18.000000
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
[]切片方法
按照索引实现行选择或列选择或区块选择
print(data[1:5]) # 行选择fecha rnd_1 rnd_2 rnd_3
1 2012-04-11 14 7 4
2 2012-04-12 2 17 3
3 2012-04-13 9 5 11
4 2012-04-14 17 10 16
print(data[[‘rnd_1’, ‘rnd_2’]]) # 列选择
rnd_1 rnd_2
0 13 19
1 14 7
2 2 17
3 9 5
4 17 10
5 16 5
6 13 4
7 10 9
8 16 18
9 12 5
区块选择， 对于多列选择，不能像行选择时一样使用1：5这样的方法来选择
print(data[:5][[‘rnd_1’, ‘rnd_2’]])
rnd_1 rnd_2
0 13 19
1 14 7
2 2 17
3 9 5
4 17 10
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
loc
loc可以按照索引来进行行列选择，包含结尾
# 按照索引行选择， loc与第一种方法不同之处在于会把第5行也选择进去，而第一种方法只会选择到第4行为止。 print(data.loc[1:5])fecha rnd_1 rnd_2 rnd_3
1 2012-04-11 14 7 4
2 2012-04-12 2 17 3
3 2012-04-13 9 5 11
4 2012-04-14 17 10 16
5 2012-04-15 16 5 18
区块选择
print(data.loc[2:4, [‘rnd_2’, ‘fecha’]])
rnd_2 fecha
2 17 2012-04-12
3 5 2012-04-13
4 10 2012-04-14
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
loc能够选择在两个特定日期之间的数据，这两个日期必须都要在索引中
# 将行索引设定为日期 data_fecha = data.set_index('fecha') print(data_fecha.head())rnd_1 rnd_2 rnd_3
fecha
2012-04-10 13 19 10
2012-04-11 14 7 4
2012-04-12 2 17 3
2012-04-13 9 5 11
2012-04-14 17 10 16
生成两个特定日期
from pandas import datetime
fecha_1 = datetime(2012, 4, 14)
fecha_2 = datetime(2012, 4, 18)
生成切片数据
print(data_fecha.loc[fecha_1: fecha_2])
rnd_1 rnd_2 rnd_3
fecha
2012-04-14 17 10 16
2012-04-15 16 5 18
2012-04-16 13 4 4
2012-04-17 10 9 12
2012-04-18 16 18 14
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
如果没有特殊需求，强烈建议使用loc而尽量少使用[]，因为loc在对DataFrame进行重新赋值操作时会避免chained indexing问题，使用[]时编译器很可能会给出SettingWithCopy的警告。
iloc
loc是按照索引（index）的值来选取，iloc是按照索引的位置来进行选取。iloc不关心索引的具体值是多少，只关心位置是多少，所以使用iloc时方括号中只能使用数值。
# 行选择 print(data_fecha.iloc[1: 5, :])rnd_1 rnd_2 rnd_3
fecha
2012-04-11 14 7 4
2012-04-12 2 17 3
2012-04-13 9 5 11
2012-04-14 17 10 16
列选择
print(data_fecha.iloc[:,[1,2]].head())
rnd_2 rnd_3
fecha
2012-04-10 19 10
2012-04-11 7 4
2012-04-12 17 3
2012-04-13 5 11
2012-04-14 10 16
切片选择
print( data_fecha.iloc[[1,3,6],[0,2]])
rnd_1 rnd_3
fecha
2012-04-11 14 4
2012-04-13 9 11
2012-04-16 13 4
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
at
at的使用方法与loc类似，但是比loc有更快的访问数据的速度，只能访问单个元素，不能访问多个元素。
iat
iat对于iloc的关系就像at对于loc的关系，是一种更快的基于索引位置的选择方法，同at一样只能访问单个元素。
ix
以上几种方法都要求查询的秩在索引中，或者位置不超过长度范围，而ix允许你得到不在DataFrame索引中的数据。
选取或者删除某列含有特殊数值的行
import pandas as pd
import numpy as np
a=np.array([[1,2,3],[4,5,6],[7,8,9]])
df1=pd.DataFrame(a,index=[‘row0’,‘row1’,‘row2’],columns=list(‘ABC’))
print(df1)
df2=df1.copy()
#选取某列含有特定数值的行
选取df1中A列包含数字1的行
df1=df1[df1[‘A’].isin([1])]
#通过~取反，选取不包含数字1的行
df2=df2[~df2[‘A’].isin([1])]
print(df1)
print(df2)
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
结果：
A B C row0 1 2 3 row1 4 5 6 row2 7 8 9A B C
row0 1 2 3
A B C
row1 4 5 6
row2 7 8 9
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
选取或者删除某行含有特殊值的列
import pandas as pd
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df2 = pd.DataFrame(a, index=[‘row0’, ‘row1’, ‘row2’], columns=list(‘ABC’))
print(df2)
#选取某行含有特定数值的列
cols=[x for i,x in enumerate(df2.columns) if df2.iat[0,i]==3]
#利用enumerate对row0进行遍历，将含有数字3的列放入cols中
print(cols)
#df2=df2[cols] 选取含有特定数值的列
df2=df2.drop(cols,axis=1) #利用drop方法将含有特定数值的列删除
print(df2)
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
结果：
A B C
row0 1 2 3
row1 4 5 6
row2 7 8 9
['C']
A B
row0 1 2
row1 4 5
row2 7 8
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9