demo.py(DataFrame切片,DataFrame取行取列):
# coding=utf-8
import numpy as np
import pandas as pd
# 创建DataFrame
df = pd.DataFrame(np.arange(12, 60).reshape((12, 4)), columns=["WW", "XX", "YY", "ZZ"])
print(df)
'''
WW XX YY ZZ
0 12 13 14 15
1 16 17 18 19
2 20 21 22 23
3 24 25 26 27
4 28 29 30 31
5 32 33 34 35
6 36 37 38 39
7 40 41 42 43
8 44 45 46 47
9 48 49 50 51
10 52 53 54 55
11 56 57 58 59
'''
# pandas取行或者列的注意点
# - 方括号写数组,表示取行,对行进行操作
# - 写字符串,表示取列索引,对列进行操作
# 取行
print(df[:5]) # 取前5行
'''
WW XX YY ZZ
0 12 13 14 15
1 16 17 18 19
2 20 21 22 23
3 24 25 26 27
4 28 29 30 31
'''
# 取一列
print(df["YY"])
'''
0 14
1 18
2 22
3 26
4 30
5 34
6 38
7 42
8 46
9