如何绕过熊猫过滤txt文档,其中一行包含字符串,整数和浮点数

时间:2020-06-18 14:12:01

标签: python pandas string filter

我的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命令为我提供了我喜欢的正确格式。但是我缺少哪一部分?

0 个答案:

没有答案