Python学习——将TXT文件读入列表或numpy数组

最新推荐文章于 2022-06-14 16:44:54 发布

卖萌的大米

最新推荐文章于 2022-06-14 16:44:54 发布

阅读量8.7k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： Python

本文链接：https://blog.youkuaiyun.com/qq_41938858/article/details/89401108

Python 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种从TXT文件读取数据并将其转换为Python列表和NumPy数组的方法。通过详细解释代码，展示了如何处理特定格式的TXT文件，包括去除不必要的字符和正确分割数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 将txt文件读入列表
def txt_to_list(path):
	"""
	path:txt文件路径
	返回:列表
	列表内容为：['1024-1.xyz', '1024-2.xyz', ...]共二百个元素
	""
    with open(path)as txt:
        cont = txt.read()
        # 去除第一个字符 [ 和最后一个字符 ]
        newcont = cont[1:] + cont[:-1]	
        # print(newcont)
        list_cont = []
        for i in range(200):
        	# 当取到最后一个元素时，索引用-1表示
        	# 不然会出错，我也不知道为什么
            if i == 199:
                i = -1
            # 以","分隔，并去除首尾空格，写入列表alph
            alph = newcont.split(',')[i].strip()
            # print(alph)

            alph = alph[1:-1]
            list_cont.append(alph)

    return list_cont

# 将txt文件读入numpy数组
def txt_to_numpy(filename, row, colu):
	"""
	filename:是txt文件路径
	row:txt文件中数组的行数
	colu:txt文件中数组的列数
	"""
    file = open (filename)
    lines = file.readlines()
    # print(lines)
	# 初始化datamat
    datamat = np.zeros((row, colu))

    row_count = 0

    for line in lines:
    	# 写入datamat
        line = line.strip().split(' ')
        datamat[row_count,:] = line[:]
        row_count += 1
    return datamat