将函数应用于以索引和列名作为参数的数据框

Question

考虑一个空数据框 (df)，其中国家作为索引，状态作为列。

df = pd.DataFrame({'Country': ['Albania', 'France', 'Mexico', 'Russia'], 
                               'On': [0, 0, 0, 0],
                               'Off': [0, 0, 0, 0]}).set_index('Country')

如何将函数应用于数据框中的每个单元格，并将索引名称和列名称作为函数的参数？我想用可以使它工作的语法替换 row(x) 和 column(x)。

df = df.apply(lambda x : function(row(x), column(x)))

谢谢

Answer 1

你可以这样做

def fct(row):
    # You can get index with row.name
    print(row.name)
    
    # You can set separately row["On"] and row["Off"]
    print(row["On"])
    print(row["Off"])
    row["On"] = row["On"] + 1
    row["Off"] = row["Off"] + 2
    return row
    
    
dff = df.apply(fct, axis=1)

编辑：如果您确实需要使用依赖于行和列的函数，这个解决方案是否适合您？

def f2(row, col):
    # ...

def fct(row):
    # You can get index with row.name
    row_name = row.name
    for column_name in row.index.values:
        row[v] = f2(row_name, column_name)
    return row
    
    
dff = df.apply(fct, axis=1)