如何移除Python列表变量中每组向量外层的单引号?
问题原因
你用pandas读取txt时,第二列内容是空格分隔数值、包裹在方括号里的字符串,pandas默认会将其识别为字符串类型存入列表,所以你看到外层带单引号。
解决方案
这里提供两种实现思路:
方案1:纯字符串处理(无需额外依赖)
直接对读取到的字符串做裁剪、拆分、类型转换即可,代码如下:
import pandas as pd grad = [] vectors = [] with open('output.txt', 'r') as result: df = pd.read_csv(result, header=None) for row in range(len(df)): grad.append(df.iloc[row, 0]) vec_str = df.iloc[row, 1] # 去除首尾空白 -> 去掉首尾方括号 -> 按空格拆分得到数值字符串 -> 转为float类型 vec = [float(num) for num in vec_str.strip()[1:-1].split()] vectors.append(vec)
方案2:用ast模块做字面量转换(兼容性更强)
如果后续向量格式有变化(比如出现多空格、科学计数法等),可以用Python标准库的ast模块做安全的字面量转换,代码如下:
import pandas as pd import ast grad = [] vectors = [] with open('output.txt', 'r') as result: df = pd.read_csv(result, header=None) for row in range(len(df)): grad.append(df.iloc[row, 0]) vec_str = df.iloc[row, 1].strip() # 把数值之间的空格替换为逗号,再直接转成Python列表 vec = ast.literal_eval(vec_str.replace(' ', ',')) vectors.append(vec)
两种方案运行后得到的vectors变量就是你期望的嵌套数值列表格式。
内容的提问来源于stack exchange,提问作者user16836078
相关产品推荐
相关产品推荐

