从csv模块中的pandas中的writerow()等于什么?

我目前正在为脚本制作不同的函数,并且我有一个可以与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

任何帮助将不胜感激!

dong137 回答:从csv模块中的pandas中的writerow()等于什么?

您可以使用

df = pd.read_csv('sample.csv',header=None)
good_nums = [3357,210]
df.loc[df['numbercolumn'].isin(good_nums)].to_csv('output.csv',index=False)

这也更快,因为您不必通过循环逐行编写。.

本文链接:https://www.f2er.com/3121312.html

大家都在问