python - Python中[]和[[]]的区别

Question

我正在 Coursera 上一门名为“使用 Python 进行数据分析”的类(class)。我是 Python 的新手。我对 C 和 MATLAB 有一些经验。这就是为什么我没有遇到那么多问题，除了一件事。
首先请看下面的代码。

import pandas as pd

#path of data
path = 'https://cf-courses-data.s3.us.cloud-object-storage.appdomain.cloud/IBMDeveloperSkillsNetwork-DA0101EN-SkillsNetwork/labs/Data%20files/automobileEDA.csv'
df = pd.read_csv(path)

from sklearn.linear_model import LinearRegression

lm = LinearRegression()

X = df[['highway-mpg']] #'highway-mpg' is a column in the dataframe
Y = df['price'] #'price' is a column in the dataframe

lm.fit(X,Y)

这里同时定义 X他们使用了双方括号 [[]]在 Y 的情况下二手 [] .我知道方括号用于在 Python 中列出并且嵌套列表是可能的。但它似乎不是一个嵌套列表。你能区分它们吗？

Answer 1

一个对数据帧进行切片并返回一个子数据帧，另一个只选择一列并将其作为 Series :

如果您将字符串列表传递给方括号 [['a', 'c']] ，您将获得 DataFrame其中仅包含您询问的列

如果您将字符串传递给方括号 ['a'] ，您将获得 只有 Series这对应于你给的名字

例子

df = pd.DataFrame([
    {'a': 1, 'b': 2, 'c': 4},
    {'a': 3, 'b': 4, 'c': 6},
])

print(df['a'])
# That is a Series
0    1
1    3
Name: a, dtype: int64
# -------------------------------------------
print(df[['a']])
# That is a DataFrame
   a
0  1
1  3
# -------------------------------------------
print(df[['a', 'c']])
# That is a DataFrame
   a  c
0  1  4
1  3  6