KeyError：在熊貓中創建新列時

Question

我正在處理數據集，當我發現差異后嘗試創建新列時，我得到了KeyError：“已過濾”

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

d = {'col1': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 'col2': [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]}
df = pd.DataFrame(data=d)
fig, ax = plt.subplots(2, figsize=(8,8))

df['col2'].diff().plot(ax=ax[0])

cutoff = 3
df['filtered'] = df.loc[df['col2'].diff().abs() > cutoff]
df.plot(ax=ax[1])

我曾經像這樣創建新列(df['filtered'] = some operation) ，但是在這種情況下它給出了KeyError: 'filtered' 。 感謝您的幫助。

Answer 1

您需要將倒數第二行替換為：

df['filtered'] = df.loc[df['col2'].diff().abs() > cutoff, 'col2']

假設您要獲取“ col2”的過濾版本。 如@RafaelC所述，當前的.loc []操作將返回行過濾器適用的所有列（在這種情況下為2），因此會出現錯誤。