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