我的txt文件格式如下:
501NA NA 1 9.517 6.338 0.776
502NA NA 2 2.683 7.229 0.642
503NA NA 3 6.856 9.313 0.543
504NA NA 4 9.412 3.246 0.808
505NA NA 5 1.994 2.141 0.620
506NA NA 6 3.571 9.574 0.575
我有熊猫可以阅读txt文件,对此我感到非常高兴。但是,当我尝试根据条件对其进行过滤时,它说不能。我希望大熊猫以它输入的确切格式吐出数据……基本上将其输出为txt字符串。
这是我的代码:
import pandas as pd
data=pd.read_csv("blockbig2.gro",sep= "\s+", header= None, keep_default_na=False)
data.columns = ['id', 'NA','index','x' ,'y','z']
print(data)
equation_x = ((data.x)-5)**2
equation_y = ((data.y)-5)**2
eq = equation_x + equation_y
data[eq<=24].to_txt('step1.txt',float_format = "%.3f", index = False ,header = False )
print命令为我提供了我喜欢的正确格式。但是我缺少哪一部分?