python-3.x - 将列添加到数据框(Pandas)时出现键错误

Question

Pandas DataFrame 并不真正接受添加第二列，我无法真正解决该问题。我正在尝试显示移动平均线。该代码仅适用于第一个 (MA_9)，并且在我尝试添加其他 MA (MA_20) 时立即出现错误。

在这种情况下是否不可能添加多于一列？

代码:

import numpy as np
import pandas as pd
import pandas_datareader as pdr
import matplotlib.pyplot as plt


symbol = 'GOOG.US'
start = '20140314'
end = '20180414'
google = pdr.DataReader(symbol, 'stooq', start, end)
print(google.head())

google_close = pd.DataFrame(google.Close)
print(google_close.last_valid_index)
google_close['MA_9'] = google_close.rolling(9).mean()
google_close['MA_20'] = google_close.rolling(20).mean()
# google_close['MA_60'] = google_close.rolling(60).mean()
# print(google_close)
plt.figure(figsize=(15, 10))
plt.grid(True)

# display MA's
plt.plot(google_close['Close'], label='Google_Cls')
plt.plot(google_close['MA_9'], label='MA 9 day')
plt.plot(google_close['MA_20'], label='MA 20 day')
# plt.plot(google_close['MA_60'], label='MA 60 day')
plt.legend(loc=2)
plt.show()

Answer 1

请按如下方式更新您的代码，然后它应该可以工作:

google_close['MA_9'] = google_close.Close.rolling(9).mean()
google_close['MA_20'] = google_close.Close.rolling(20).mean()

最初，Close 只有一列数据，因此您的旧代码 google_close['MA_9'] = google_close.rolling(9).mean() 可以工作，但在这行代码之后，现在它有了两列，所以它不知道您想要表示哪些数据。因此，使用您想要表示的数据的列详细信息进行更新，它可以工作 google_close['MA_20'] = google_close.Close.rolling(20).mean()