将文本文件中的特定列制作成列表

Question

我有一个看起来像这样的文本文件：

TITPLE="SurfaceData"
VARIABLES="X", "Y", "Z", "Cutoff Rigidity", "Injected Particle Number"
ZONE N=30, E=25, DATAPACKING=POINT, ZONETYPE=FEQUADRILATERAL
0.000000e+00 0.000000e+00 6.371000e+06  4.843809e-01  2.001000e+03
0.000000e+00 0.000000e+00 6.371000e+06  4.635282e-01  2.022500e+03
0.000000e+00 0.000000e+00 6.371000e+06  1.158746e+00  1.984500e+03
0.000000e+00 0.000000e+00 6.371000e+06  1.153130e+00  1.960500e+03
0.000000e+00 0.000000e+00 6.371000e+06  4.873921e-01  1.986500e+03
1 2 7 6
2 3 8 7
3 4 9 8
4 5 10 9

在 python 3 中，我想将前 3 列数据转换为 3 个单独的列表，并忽略文件中的所有其他列和数据。换句话说，我想将以下三列转换为三个列表：

0.000000e+00 0.000000e+00 6.371000e+06
0.000000e+00 0.000000e+00 6.371000e+06 
0.000000e+00 0.000000e+00 6.371000e+06  
0.000000e+00 0.000000e+00 6.371000e+06  
0.000000e+00 0.000000e+00 6.371000e+06

其中每一列代表一个单独的列表。当文件中没有其他数据时，我已经看到了执行此操作的方法，但我找不到忽略数据上方的文本和该数据下方的整数集的方法。有人有解决办法吗？

Answer 1

你可以使用pandas，它也修复了科学记数法：

import pandas as pd

df = pd.read_csv('filename.txt', delim_whitespace=True, header=None, skiprows=3, skipfooter=4, engine='python')

输出df：

	2	3	4
0	6371000.0	0.4843809	2001.0
1	6371000.0	0.4635282	2022.5
2	6371000.0	1.1587459999999998	1984.5
3	6371000.0	1.15313	1960.5
4	6371000.0	0.48739209999999994	1986.5

只保留前三列：df = df[[0,1,2]]

将列转换为列表：df[0].tolist()

将文本文件中的特定列制作成列表

Making specific columns in a text file into a list

python

file-io