dataframe行列查詢

一、可以通過df.shape查詢行列數

import pandas as pd

df = pd.DataFrame([[1,2,3], [7,5,7], [9,7,4], [8,4,6]])
# 維度查詢
>>> df.shape
(4, 3)

# 獲取行數
>>> df.shape[0]
4

# 獲取列數
>>> df.shape[1]
3

二、查詢行列名

# 我們可以先修改一下行列名
df.columns = ['col1', 'col2', 'col3']
df.index = ['i1', 'i2', 'i3', 'i4']
>>> df
    col1  col2  col3
i1     1     2     3
i2     7     5     7
i3     9     7     4
i4     8     4     6

# 獲取列名
>>> df.columns
Index(['col1', 'col2', 'col3'], dtype='object')
# 獲取行名
>>> df.index
Index(['i1', 'i2', 'i3', 'i4'], dtype='object')

三、獲取某行或某列

# 獲取某行
df.loc['i1']
col1    1
col2    2
col3    3
Name: i1, dtype: int64

# 獲取某幾行
>>> df.loc['i1':'i3']
    col1  col2  col3
i1     1     2     3
i2     7     5     7
i3     9     7     4

# 獲取某列
>>> df['col1']
i1    1
i2    7
i3    9
i4    8
Name: col1, dtype: int64

# 獲取某幾列
>>> df[['col1','col2']]
    col1  col2
i1     1     2
i2     7     5
i3     9     7
i4     8     4

# 也可用查詢行的方法獲取列
>>> df.loc[:, ['col1', 'col2']]
    col1  col2
i1     1     2
i2     7     5
i3     9     7
i4     8     4