列表的“len”是否会在 for 循环的每次迭代中重新计算？

Question

我需要删除日期在 2000 年之前的数据框的所有列。

一般做法是：

columnstokeep = list(DF) #gives me the column names

    for i in range(len(columnstokeep)): #get rid of dates before year 2000
        if int(columnstokeep[i][:4])<2000:
            columnstokeep.remove(columnstokeep[i])

DF = DF[columnstokeep]  #the new dataframe

我一直遇到列表索引超出范围错误。

这是因为每次我删除列表中的一个元素时 len, in range(len(columnstokeep)) 都会发生变化吗？还是 range(len(columnstokeep)) 在循环期间保持相同的值？

这是数据框

谢谢

Answer 1

您可以在列上使用pd.to_datetimes 轻松完成此操作，然后选择大于2000 的列。

# Create Example Data
frame = pd.DataFrame({
    '1998-1-1': ['foo'],
    '1999-1-1': ['bar'],
    '2000-1-1': ['spam'],
    '2001-1-1': ['eggs']
})

# Select columns which are after 2000
frame.loc[:,pd.to_datetime(frame.columns) >= '2000']

输出：

  2000-1-1 2001-1-1
0     spam     eggs

Answer 2

您对问题根源的看法是正确的。但我不认为范围会被重新计算。但是由于您从列表中删除第一个值i 将超出剩余的columnstokeep。我添加了一些打印以更清楚地显示问题：

years = range(1990,2010) 
columnstokeep=[]
#The column names kind of
for i in years:
    columnstokeep.append(str(i)+'-01')

##This shows the error comment this
for i in range(len(columnstokeep)-1): #get rid of dates before year 2000
    print(i,columnstokeep[i])#It prints every second year while in 199X

    if int(columnstokeep[i][:4])<2000:
        columnstokeep.remove(columnstokeep[i])

相反，您可以从结束迭代到开始......

for i in range(len(columnstokeep)-1,-1,-1): #get rid of dates before year 2000
    print(i,columnstokeep[i])#It prints every second year while in 199X

    if int(columnstokeep[i][:4])<2000:
        columnstokeep.remove(columnstokeep[i])



#DF = DF[columnstokeep]  #the new dataframe
print(columnstokeep)

输出：

['2000-01', '2001-01', '2002-01', '2003-01', '2004-01', '2005-01', '2006-01', '2007-01', '2008-01', '2009-01']