如何在 Pandas 数据框的列中添加空白单元格？

Question

我是 Python 的初学者，并且通过 Pandas 使用数据框。我正在尝试使用xml.dom.minidom 从 XML 文件中提取表格到 Excel 文件中。这是原始表格的样子（注意“自行车”下的黑色条目）：

VEHICLE   BRAND
Car       Mercedes
Bike      Kawasaki
          Ducati
Truck     Ram

我正在尝试从给定的 XML 文件中提取此表：

<Info_Collection>
    <Info car="Car">
        <V_Collection>
            <Brand type="Mercedes"/>
        </V_Collection>
    </Info>
    <Info car="Bike">
        <V_Collection>
            <Brand type="Kawasaki"/>
            <Brand type="Ducati"/>
        </V_Collection>
    </Info>
    <Info car="Truck">
        <V_Collection>
            <Brand type="Ram"/>
        </V_Collection>
    </Info>
</Info_Collection>

这是我正在使用的代码：

def main():
  x1=[]
  x2=[]
  doc = xml.dom.minidom.parse('xml_file')
  t1 = doc.getElementsByTagName("Info")
  t2 = doc.getElementsByTagName("Brand")
  for a in t1:
     x1.append(tb.getAttribute("car"))
  for a in t2:
     x2.append(tb.getAttribute("type"))
  while len(x1) != len(x2): 
     x1.append("")
  boDF = pd.DataFrame({'VEHICLE': x1, 'BRAND':x2})
  boDF.to_excel(writer, sheet_name='Sheet1', index=0, startrow=1)
  writer.save()

if __name__ == "__main__":
  main()

运行后，输出表如下：

VEHICLE   BRAND
Car       Mercedes
Bike      Kawasaki
Truck     Ducati
          Ram

谁能帮我弄清楚如何在“自行车”和“卡车”之间插入一个空格？我尝试同时运行两个 for 循环并比较它们的长度以查看它们是否相等，当它们不相等时，将在第一列中添加一个空格。但是，我无法让它工作。我知道我的代码中的 while 循环在第一列的末尾添加了一个空格，但我不知道如何在列内的任何位置添加。

Answer 1

见下文。

这个想法是创建一个通常与 Excel 关联的 csv 文件。 XML 解析由内置的 python XML 解析库“ElementTree”完成。当你双击 csv 文件时，它通常会被 Excel 打开，你会得到你正在寻找的表格。

注意您发布的 XML 不是有效的 XML，必须修复（v_Collection Vs V_Collection）

import xml.etree.ElementTree as ET

xml = '''<Info_Collection>
    <Info car="Car">
        <V_Collection>
            <Brand type="Mercedes"/>
        </V_Collection>
    </Info>
    <Info car="Bike">
        <V_Collection>
            <Brand type="Kawasaki"/>
            <Brand type="Ducati"/>
        </V_Collection>
    </Info>
    <Info car="Truck">
        <V_Collection>
            <Brand type="Ram"/>
        </V_Collection>
    </Info>
</Info_Collection>'''

data = [['VEHICLE', 'BRAND']]
root = ET.fromstring(xml)
info_list = root.findall('.//Info')
cars = set()
for info_entry in info_list:
    car = info_entry.attrib['car']
    for brand in [brand.attrib['type'] for brand in info_entry.findall('.//Brand')]:
        data.append([car if car not in cars else '', brand])
        cars.add(car)
with open('cars.csv', 'w') as out:
    for entry in data:
        out.write(','.join(entry) + '\n')

输出（cars.csv）

VEHICLE,BRAND
Car,Mercedes
Bike,Kawasaki
,Ducati
Truck,Ram