我有这样一个正方形矩阵:
[[0, 516, 226, 853, 1008, 1729, 346, 1353, 1554, 827, 226, 853, 1729, 1008],
[548, 0, 474, 1292, 1442, 2170, 373, 1801, 1989, 1068, 474, 1292, 2170, 1442],
[428, 466, 0, 1103, 1175, 1998, 226, 1561, 1715, 947, 0, 1103, 1998, 1175],
[663, 1119, 753, 0, 350, 1063, 901, 681, 814, 1111, 753, 0, 1063, 350],
[906, 1395, 1003, 292, 0, 822, 1058, 479, 600, 1518, 1003, 292, 822, 0],
[1488, 1994, 1591, 905, 776, 0, 1746, 603, 405, 1676, 1591, 905, 0, 776],
[521, 357, 226, 1095, 1167, 1987, 0, 1552, 1705, 1051, 226, 1095, 1987, 1167],
[1092, 1590, 1191, 609, 485, 627, 1353, 0, 422, 1583, 1191, 609, 627, 485],
[1334, 1843, 1436, 734, 609, 396, 1562, 421, 0, 1745, 1436, 734, 396, 609],
[858, 1186, 864, 1042, 1229, 1879, 984, 1525, 1759, 0, 864, 1042, 1879, 1229],
[428, 466, 0, 1103, 1175, 1998, 226, 1561, 1715, 947, 0, 1103, 1998, 1175],
[663, 1119, 753, 0, 350, 1063, 901, 681, 814, 1111, 753, 0, 1063, 350],
[1488, 1994, 1591, 905, 776, 0, 1746, 603, 405, 1676, 1591, 905, 0, 776],
[906, 1395, 1003, 292, 0, 822, 1058, 479, 600, 1518, 1003, 292, 822, 0]]
我需要同时删除a1
{a3
索引列和行。我该怎么做?整洁的方式是什么
注意,我需要得到另一个平方矩阵。应删除同一索引中的行和列。还要注意的是,删除行/列时,索引会移位。要么我也需要转换,例如a1
,a2
,a3
,要么做一些更聪明的事情
方阵:
[[10,11,12,13],
[14,15,16,17],
[18,19,20,21],
[22,23,24,25]]
删除第一个和第三个索引,结果是:
[[10,12],
[18,20]]
如果您对其他软件包持开放态度,pandas可以简化:
更新:输出样本数据上的代码
如果只需要numpy并假设数组始终为平方:
在numpy中尝试此方法
np.ix_
为您创建一个网格网格,用于索引numpy数组的列和行。只需在方阵中的行范围和要删除的行/列的索引列表之间取set.difference
,即可创建索引列表-编辑:下面添加了要删除的方阵10000x1000和~500行/列的基准测试结果。(MacBookPro 13)
相关问题 更多 >
编程相关推荐