python-3.x - 如何根据列值创建和弦图矩阵:
问题描述
假设我有一个数据框,其中包含以下格式的数据。
UID | Name | ID
----------------
1 | ABC | IM-1
2 | XYZ | IM-2
3 | XYZ | IM-2
4 | PQR | IM-3
5 | PQR | IM-4
6 | PQR | IM-5
7 | XYZ | IM-5
8 | ABC | IM-5
我需要创建一个输入和弦图代码的矩阵。这需要以下格式的输出:
(array([[0,1,1,1],
[1,1,1,0],
[1,1,0,2]]),['ABC','XYZ','PQR'])
注意:在此示例中,-“名称”在列表中是有限的(即 ABC、XYZ 或 PQR)-“ID”在记录之间共享-第四列是独立的记录数(例如 ABC 是部分IM-1和PQR在IM-4和IM-5中出现两次 - 矩阵的其他成员是基于 ID 的名称之间的联系(例如IM-5,增加PQR-XYZ、XYZ的值-PQR , PQR-ABC , ABC-PQR , XYZ-ABC & ABC-XYZ ) - 目标是为“名称”字段之间的连接创建一个和弦图
我知道这是一本好书。在此先感谢您的帮助。
解决方案
更新了我的答案,但方法基本相同。将数据解析为数据框,进行内部连接以ID
获取通过共享 common 链接的名称对ID
。然后将此边列表转换为邻接矩阵。最后,为了获得“悬空”边缘,即ID
只出现一次(在更新的答案中添加),并将它们的计数按相应的Name
.
#!/usr/bin/env python
"""
Create adjacency matrix from a dataframe, where edges are implicitly defined by shared attributes.
Answer to:
https://stackoverflow.com/questions/57849602/how-to-create-the-matrix-for-chord-diagram-based-on-coloumn-value
"""
import numpy as np
import pandas as pd
from collections import Counter
def parse_data_format(file_path):
# read data skipping second line
df = pd.read_csv(file_path, sep='|', skiprows=[1])
# strip whitespace from column names
df = df.rename(columns=lambda x: x.strip())
# strip whitespace from values
df_obj = df.select_dtypes(['object'])
df[df_obj.columns] = df_obj.apply(lambda x: x.str.strip())
return df
def get_edges(df):
"""Get all combinations of 'Name' that share a 'ID' value (using an inner join)."""
inner_self_join = df.merge(df, how='inner', on='ID')
excluding_self_pairs = inner_self_join[inner_self_join['UID_x']!=inner_self_join['UID_y']]
edges = excluding_self_pairs[['Name_x', 'Name_y']].values
return edges
def get_adjacency(edges):
"Convert a list of 2-tuples specifying source and target of a connection into an adjacency matrix."
order = np.unique(edges)
total_names = len(order)
name_to_idx = dict(list(zip(order, range(total_names))))
adjacency = np.zeros((total_names, total_names))
for (source, target) in edges:
adjacency[name_to_idx[source], name_to_idx[target]] += 1
return adjacency, order
def get_dangling_edge_counts(df):
# get IDs with count 1
counts = Counter(df['ID'].values)
singles = [ID for (ID, count) in counts.items() if count == 1]
# get corresponding names
names = [df[df['ID']==ID]['Name'].values[0] for ID in singles]
# convert into counts
return Counter(names)
if __name__ == '__main__':
# here we read in the data as a file buffer;
# however, normally we would hand a file path to parse_data_format instead
import sys
if sys.version_info[0] < 3:
from StringIO import StringIO
else:
from io import StringIO
data = StringIO(
"""UID | Name | ID
----------------
1 | ABC | IM-1
2 | XYZ | IM-2
3 | XYZ | IM-2
4 | PQR | IM-3
5 | PQR | IM-4
6 | PQR | IM-5
7 | XYZ | IM-5
8 | ABC | IM-5
"""
)
df = parse_data_format(data)
edges = get_edges(df)
adjacency, order = get_adjacency(edges)
print(adjacency)
# [[0. 1. 1.]
# [1. 0. 1.]
# [1. 1. 0.]]
print(order)
# ['ABC' 'PQR' 'XYZ']
dangling_edge_counts = get_dangling_edge_counts(df)
print(dangling_edge_counts)
# Counter({'PQR': 2, 'ABC': 1})
last_column = np.zeros_like(order, dtype=np.int)
for ii, name in enumerate(order):
if name in dangling_edge_counts:
last_column[ii] = dangling_edge_counts[name]
combined = np.concatenate([adjacency, last_column[:, np.newaxis]], axis=-1)
print(combined)
#[[0. 1. 1. 1.]
# [1. 0. 1. 2.]
# [1. 1. 2. 0.]]
推荐阅读
- node.js - Mongodb检查字符串数组中的任何字符串是否存在另一个字符串
- python - Python:如何从包含列表作为值的字典列表中制作 DataFrame
- r - 在R中为年龄范围创建年龄分层表
- javascript - 如何在javascript中按日期名称对数组进行排序?
- asp.net-core - AuditIgnore 对某些功能
- python - 如何将 CUDA 与 vaex(一个 Python 库)一起使用
- c++ - 将 char 向量转换为 size_t 的 C++ 安全方法
- eclipse-hono - 如何在部署在 K8s 集群上的 Eclipse Hono 中进行服务配置?
- typescript - 根据 serenity-platform 中的条件设置必填/不需要的字段
- blockchain - 使用 ipfs 和区块链的文件搜索和实现