1. 程式人生 > 其它 >pamdas-DataFrame使用索引或切片獲取行和列

pamdas-DataFrame使用索引或切片獲取行和列

技術標籤:pandasPython基礎知識pandas

程式碼示例:

import pandas as pd

df = pd.read_csv('test.csv')
print(df)
'''
列印:
   userId  score  age
0       1     45   18
1       2     65   19
2       3     58   17
3       4     92   16
4       5     78   18
'''


#獲取列
print(type(df.score))    #列印:<class 'pandas.core.series.Series'>
print(df.score)
'''
列印:
0    45
1    65
2    58
3    92
4    78
Name: score, dtype: int64
'''
print(type(df['score']))    #列印:<class 'pandas.core.series.Series'>
print(df['score'])
'''
列印:
0    45
1    65
2    58
3    92
4    78
Name: score, dtype: int64
'''
print(type(df[['score']]))    #列印:<class 'pandas.core.frame.DataFrame'>
print(df[['score']])
'''
列印:
   score
0     45
1     65
2     58
3     92
4     78
'''
# 獲取多列
print(df[['age','score']])
'''
列印:
   age  score
0   18     45
1   19     65
2   17     58
3   16     92
4   18     78
'''


#獲取行。df[] 可以使用切片,獲取的是行,可以使用bool型別列表,獲取的也是行
print(df[0:1])
'''
列印:
   userId  score  age
0       1     45   18
'''
df.index = list("12345")
print(df["1":"1"])
'''
列印:
   userId  score  age
1       1     45   18
'''
print(df[[True,False,True,False,True]])
'''
輸入的bool陣列長度必須跟DataFrame的行數一樣
列印:
   userId  score  age
1       1     45   18
3       3     58   17
5       5     78   18
'''
print(df[df['age']>18])
'''
列印:
   userId  score  age
2       2     65   19
'''