KOTA ITO

Generation

generate functionWed, 07 Dec 2022

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

# input: pd.DataFrame # output: np.array def adjacent_matrix(df): # adjacent matrix adjacent_matrix = np.zeros((len(df),len(df))) # fill adjacent matrix for idx, row in df.iterrows(): adjacent_matrix[idx][row.to_numpy()] = 1 return adjacent_matrix

Questions about programming?Chat with your personal AI assistant