问题描述
我是第一次这样做,到目前为止,已经设置了一个简单的脚本来从API提取2列数据
数据通过,并且可以使用打印命令看到它
现在正在尝试将其写入CSV并设置下面的代码以创建文件,但我不知道如何:
1。删除每个数据行之间的空白行
2。在我想成为“”的数据中添加定界符
3。如果IP之类的值为空白,则仅显示“”。
我搜索并尝试了各种示例,但都遇到错误
我成功编写CSV的代码段是
import requests
import csv
import json
# Make an API call and store response
url = 'https://api-url-goes-here.com'
filename = "test.csv"
headers = {
'accept': 'application/json',}
r = requests.get(url,headers=headers,auth=('User','PWD'))
print(f"Status code: {r.status_code}")
#Store API response in a variable
response_dict = r.json()
#Open a File for Writing
f = csv.writer(open(filename,"w",encoding='utf8'))
# Write CSV Header
f.writerow(["Computer_Name","IP_Addresses"])
for computer in response_dict["advanced_computer_search"]["computers"]:
f.writerow([computer["Computer_Name"],computer["IP_Addresses"]])
我得到的CSV输出如下:
计算机名称,IP地址
HYDM002543514,
HYDM002543513,10.93.96.144-AirPort-zh-
HYDM002544581,192.168.1.8-AirPort-en1 / 10.93.224.177- GlobalProtect-gpd0
HYDM002544580,10.93.80.101-以太网-en0
HYDM002543515,192.168.0.6-AirPort-en0 / 10.91.224.58- GlobalProtect-gpd0
CHAM002369458,10.209.5.3-以太网-en0
CHAM002370188,192.168.0.148-AirPort-en0 / 10.125.91.23- GlobalProtect-gpd0
MacBook-Pro,
我尝试添加
csv.writer(f,delimiter =' ',quotechar =',',quoting=csv.QUOTE_MINIMAL)
在f = csv.writer行之后,但会产生错误:
TypeError:参数1必须具有“写入”方法
我确信它很简单,但是找不到在我拥有的代码中实现的正确解决方案。任何帮助表示赞赏。
此外,文件会自动关闭吗?一些示例建议使用类似f.close()的方法,但这会导致错误。我需要吗?该文件似乎可以按原样创建。
解决方法
我建议您使用pandas软件包编写.csv
文件,这是用于数据分析的最常用的软件包。
针对您的问题:
import requests
import csv
import json
import pandas
# Make an API call and store response
url = 'https://api-url-goes-here.com'
filename = "test.csv"
headers = {
'accept': 'application/json',}
r = requests.get(url,headers=headers,auth=('User','PWD'))
print(f"Status code: {r.status_code}")
#Store API response in a variable
response_dict = r.json()
#collect data to build pandas.DataFrame
data = []
for computer in response_dict["advanced_computer_search"]["computers"]:
# filter blank line
if computer["Computer_Name"] or computer["IP_Addresses"]:
data.append({"Computer_Name":computer["Computer_Name"],"IP_Addresses":computer["IP_Addresses"]})
pandas.DataFrame(data=data).to_csv(filename,index=False)
如果要使用" "
来分隔值,可以在输出sep=" "
文件的最后一行中设置.csv
。但是,由于这是一个通用标准,我建议使用,
作为分隔符。还可以为DataFrame.to_csv()方法设置更多配置,您可以查看官方文档。 https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.to_csv.html
正如您在评论中所说,pandas不是标准的python软件包。您可以简单地打开一个文件,并使用手动生成的行将行写入该文件。例如:
import requests
import csv
import json
# Make an API call and store response
url = 'https://api-url-goes-here.com'
filename = "test.csv"
headers = {
'accept': 'application/json','PWD'))
print(f"Status code: {r.status_code}")
#Store API response in a variable
response_dict = r.json()
r = requests.get(url,'PWD'))
print(f"Status code: {r.status_code}")
#Store API response in a variable
response_dict = r.json()
#Open a File for Writing
f = csv.writer(open(filename,"w",encoding='utf8'))
with open(filename,mode='w') as f:
# Write CSV Header
f.write("Computer_Name,"+"IP_Addresses"+"\n")
for computer in response_dict["advanced_computer_search"]["computers"]:
# filter blank line
if computer["Computer_Name"] or computer["IP_Addresses"]:
f.write("\""+computer["Computer_Name"]+"\","+"\""+computer["IP_Addresses"]+"\"\n")
请注意,“围绕价值是通过附加\"
来建立的。\n
会在每次循环后更改新行。