如何从 Pandas 向子数据框添加新列

Question

我开始使用熊猫，所以这可能是一个非常明显的问题，但我已经为此苦苦挣扎了一段时间，但没有找到解决方案。

考虑这个数据框：

import pandas_datareader as pdr
apple = pdr.DataReader('AAPL', data_source='yahoo', 
                      start=datetime.datetime(2013, 1, 1), 
                      end=datetime.datetime(2020, 1, 1))

现在，我可以简单地向这个数据框添加一个新列：

apple['new_column'] = np.arange(apple.shape[0])

但是，如果我使用iloc提取子数据框并尝试向子数据框添加新列：

apple_2 = apple.iloc[1:5,:]
apple_2['test2'] = np.arange(4)

我收到错误消息：

<stdin>:1: SettingWithCopyWarning: 
A value is trying to be set on a copy of a slice from a DataFrame.
Try using .loc[row_indexer,col_indexer] = value instead

我做错了什么，我应该如何在熊猫中做到这一点？ 该错误建议使用.loc但我不知道如何使用它来添加新列。

Answer 1

你可以做：

apple_2 = apple.loc[:, 'High':'Close']

这将为您提供“高”和“关闭”（不关闭）之间的所有列。 但是还有其他方法可以对数据帧进行列切片。 你一个检查这个问题。

编辑：

apple_2 = apple.loc[:, 'High':'Close']
#add a new column to apple_2
apple_2['new_column'] = np.arange(apple_2.shape[0])