python - 如何循环遍历 Pandas 数据框

Question

我正在尝试循环遍历 Pandas 数据框。列表 L 包含用于指定 X 或 Y 行应从哪一行开始的值，即 (1:, 2:, 3: )。

list = [1,2,3]

for L in list:
    X = data.ix[L:, 'X':]
    Y = data.ix[L:, 'Y']     
    regressor = LinearRegression()
    regressor.fit(X, Y)
    prediction = regressor.predict([[Variable]])

尝试上述操作时出现的错误是:

TypeError: 'type' object is not iterable

Answer 1

IIUC 你可以执行以下操作:

l = [1,2,3]
results = []

for idx in l:
    X = data.ix[idx:, 'X':]
    Y = data.ix[idx:, 'Y']     
    regressor = LinearRegression()
    regressor.fit(X, Y)
    results.append(regressor.predict([[Variable]]))

但是，我不知道这里的 Variable 是什么，您也可以执行以下操作:

for df in data.iloc[::1]:
    regressor = LinearRegression()
    regressor.fit(df['X'], df['Y'])
    results.append(regressor.predict([[Variable]]))