Python中利用set对二维数组去重并转换为一维数组
发布时间：2026-10-13| 5人浏览过
本文介绍在Python中使用set对二维数组进行去重并转换为一维数组的方法。由于set无法直接处理嵌套列表，直接调用matrix.tolist()会产生二维列表导致报错。通过结合使用.A.flatten()方法将矩阵展平为一维后再转换为列表，即可成功传入set进行去重操作。
for splitValue in set(dataset[:, featureIndex].tolist()):
二维数组转一维与去重问题
首先需要明确，set是一种无序且不含重复元素的数据结构，常用于数据去重。然而，set接收的参数类型有限，若直接传入原生array可能会遇到问题。当前场景中使用的是numpy中的matrix对象，尝试直接调用matrix.tolist()发现返回值格式不符合预期：
import os
os.chdir("D:\\galaxy\\aliyunsvn\\code\\MLInAction\\dataset")
dataArr = loadDataSet("ex00.txt")
dataMat = mat(dataArr)
dataMat[:, 0].tolist()输出结果为：
[[0.036098], [0.727098],..
这种嵌套列表结构并非期望的一维形式，若直接将其放入set中会引发错误：
错误原因分析
set(dataMat[:, 0].tolist())
>>TypeError: unhashable type: 'list'经过查阅相关资料，对代码进行了调整：
解决方案与代码实现
import os
os.chdir("D:\\galaxy\\aliyunsvn\\code\\MLInAction\\dataset")
dataArr = loadDataSet("ex00.txt")
dataMat = mat(dataArr)
dataMat[:, 0].A.flatten().tolist()
>> [0.036098, 0.712386, ....最终得到了理想的一维数组格式，随后将其作为参数传入set中，实现了完美去重。
set(dataMat[:, 0].A.flatten().tolist())以上即为本文的全部内容，希望能为读者的学习提供有益参考，同时也欢迎大家继续支持本站。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
