Kenny Programming

1편에서 pandas를 이용해 xml정보를 csv엑셀로 저장하는것을 해보았다.

그러다 특정 정보에서 charge라는 요금항목을 누락해서 xml return해주는 경우가 발생해서

오류가 났다.

☆ 정상 xml 리턴

df.loc[i] = [item.arrplacenm.text, item.arrplandtime.text, item.charge.text, item.depplacenm.text,
AttributeError: 'NoneType' object has no attribute 'text'

Process finished with exit code 1

조건을 주어 에러를 방지해야하겠다.

기존의 코드는 이러했다.

for item in items.findAll("item"):

    df.loc[i] = [item.arrplacenm.text, item.arrplandtime.text, item.charge.text, item.depplacenm.text,
                 item.depplandtime.text, item.gradenm.text, item.routeid.text]

    i=i+1

에러방지를 위해 아래와 같이 추가해 주었다.

charge = "-"

for item in items.findAll("item"):

    if (item.charge != None):
        charge = item.charge.text
    else:
        charge = "-"

    df.loc[i] = [item.arrplacenm.text, item.arrplandtime.text, charge, item.depplacenm.text,
                 item.depplandtime.text, item.gradenm.text, item.routeid.text]

    i=i+1

item에 charge가 있을때만 가져오고 없을때는 -로 저장되도록 수정하고 정상적으로 작동되었다.

다른값들도 혹시나 하는 오류를 방지를 위해서 위와같이 변경해줘도 좋을것 같다.

# -*- coding: utf-8 -*-

import os
from bs4 import BeautifulSoup
from urllib.request import urlopen
import pandas as pd

output_path = "output"
output_file = "고속_test.csv"
if not os.path.exists(output_path):
    os.makedirs(output_path)

df = pd.DataFrame(columns=['arrPlaceNm', 'arrPlandTime', 'charge', 'depPlaceNm', 'depPlandTime', 'gradeNm', 'routeId']) # 엑셀 헤더정보

i=0

url = 'xml정보 주소'

data = urlopen(url).read()
soup = BeautifulSoup(data, "html.parser")
items = soup.find("items")

for item in items.findAll("item"):

    df.loc[i] = [item.arrplacenm.text, item.arrplandtime.text, item.charge.text, item.depplacenm.text,
                 item.depplandtime.text, item.gradenm.text, item.routeid.text]

    i=i+1

df.to_csv(os.path.join(output_path, output_file), encoding='euc-kr', index=False)

위와 같이 pandas를 이용해 어렵지 않게 csv로 원하는 정보를 얻을수 있었다.

하지만 여기에서 문제가 발생했다.

위와같이 YYYYMMDDHHmm의 데이터를 지수형으로 표시해주고있다.

Ctrl + C로 복사후 메모장 같은 곳에 붙여 넣어도 지수형으로 붙여 넣어진다.

위 문제가 생긴 2개의 정보에 아래와 같이 코드를 추가해 주었다. 

df['arrPlandTime'] = '="' + df['arrPlandTime'] + '"'
df['depPlandTime'] = '="' + df['depPlandTime'] + '"'

이제 다시 엑셀을 확인해보자

엑셀에도 이쁘게 잘 나오고

복사 후 메모장 등에 붙여넣어도 정확하게 표시된다.