KOTA ITO

Generation

generate functionWed, 07 Dec 2022

列が二つあるデータフレームから隣接行列を作成する。

def make_adjacent_matrix(df, col0, col1): n = df.shape[0] matrix = np.zeros((n, n)) for i, r0 in enumerate(df[col0]): for j, r1 in enumerate(df[col1]): if r0 == r1: matrix[i, j] = 1 return matrix make_adjacent_matrix(df, "from", "to")

Questions about programming?Chat with your personal AI assistant