行と列の両方に含まれるステーションのみに行列を制限

gistfile1.txt

# 行と列の両方に存在するステーション名を抽出
common_stations = set(transition_matrix.index) & set(transition_matrix.columns)

# 行と列の両方に含まれるステーションのみに行列を制限
filtered_matrix = transition_matrix.loc[common_stations, common_stations]

# filtered_matrix の各行の合計を計算
row_sums = filtered_matrix.sum(axis=1)

# 行和が0でないステーションのリストを取得
non_zero_stations = row_sums[row_sums != 0].index

# 行和が0でないステーションに対応する行と列のみを保持
filtered_matrix = filtered_matrix.loc[non_zero_stations, non_zero_stations]

# 各行の合計を新しい列として追加
filtered_matrix['Row Sum'] = row_sums

filtered_matrix.to_csv(path + 'transition_count_matrix.csv', index=True)

# 結果を表示
filtered_matrix