一、创建数据源
df = pd.DataFrame([[1, 2], [4, 5], [7, 8]],
index=['cobra', 'viper', 'sidewinder'],
columns=['max_speed', 'shield'])
df
# max_speed shield
# cobra 1 2
# viper 4 5
# sidewinder 7 8
二、仅通过方括号,提取数据
1、一个方括号,一个参数: 提取 1列数据。返回 Series
df["max_speed"]
# cobra 1
# viper 4
# sidewinder 7
# Name: max_speed, dtype: int64
df["cobra"]
""" 报错原因:一个方括号,用于列的索引,而cobra 不在列索引中 """
# Traceback (most recent call last):
# ...
# KeyError: 'cobra'
2、两个方括号,一个参数: 提取 1列数据。返回 DataFrame
df[["max_speed"