我目前正在为脚本制作不同的函数,并且我有一个可以与writerow()
完美配合的函数,但是对于其他函数,我使用了pandas,我正在尝试寻找与之等效的函数。我很确定to_csv
不是我想要的。
这是正在两个功能上测试的同一CSV。
jeffrey,192.168.1.1,example1.com,30220,internet serverice provider 1
mike,192.168.1.2,example2.com,internet service provider 1
frank,192.168.1.3,example3.com,internet service provider 1
lucy,192.168.1.4,example4.com,14619,internet service provider 2
louisa,192.168.1.5,example5.com,internet service provider 2
emily,192.168.1.6,example6.com,3357,internet service provider 3
john,192.168.1.7,example7.com,210,internet service provider 4
这是我的脚本,适用于writerow()
import csv
document= open("sample.csv")
def start_yes():
with open ('good_numbers','w') as output:
with document as file:
output_data = csv.writer(output,delimiter = ',')
reader = csv.reader(file)
list_1 = ['3357','210']
for row in reader:
if row[3] in list_1:
output_data.writerow(row)
使用sample.csv运行此脚本,这是我想要的结果。
emily,internet service provider 4
这是我尝试使用pandas模块查找与writerow()
等效的脚本。
import pandas as pd
df = pd.read_csv('sample.csv',header=None)
good_nums = ['3357','210']
bad_nums = ['30220']
maybe_nums = list(set(df[3].tolist()) - set(good_nums + bad_nums))
for asn in df:
if asn in df[3]:
asn.to_csv('output.csv',index=False)
如何使用使用熊猫的脚本获得类似这样的结果
lucy,internet service provider 2
任何帮助将不胜感激!