KOTA ITO

Generation

generate functionWed, 07 Dec 2022

pandasデータフレームから隣接行列を生成する。

import pandas as pd import numpy as np def func(df, v): matrix = df.pivot_table(index=v, columns=v, values='value', aggfunc=np.size) matrix.fillna(0, inplace=True) matrix = matrix.apply(lambda x: x/sum(x)) return matrix data = pd.DataFrame({'A':['a','a','b','b'], 'B':['c', 'd', 'c', 'd'], 'value':[1,1,1,1]}) func(data, 'A')

Questions about programming?Chat with your personal AI assistant