熊猫在循环中选择下一列标题

Question

我试图在循环遍历 pandas 数据框的列时访问下一列标题。for cols in data.columns:     if data.columns.get_loc(cols) < len(data.columns):  # skip last column of data         count = data.groupby([cols, cols+1]).size()  # create new df and how many times the two columns occur但是cols+1给我一个错误。这是因为cols返回标题名称，所以你不能 +1 一个字符串，但是在这样的循环中获取下一列标题的最佳方法是什么？

Answer 1

您可以枚举列

for col_index, cols in enumerate(data.columns): 
   if col_index+1 < len(data.columns):  # skip last column of data
        count = data.groupby([cols,data.columns[col_index+1]]).size()  # create new df and how many times the two columns occur

Answer 2

for col_idx in range(len(data.columns)-1):
    if data.columns.get_loc(data.columns[col_idx]) < len(data.columns): 
        count = data.groupby([col_idx, col_idx+1]).size()