lxml：在csv格式的列表中循环时创建多个标记

ni65a41a 于 2023-02-10 发布在其他

关注(0)|答案(1)|浏览(142)

我有这样一个csv文件：

customer, restaurant, drink, dish 
1, MC, "C1, C6", "D1, D2, D3"

我想用Python lxml模块得到这样的XML：

<Example>
<Customer> 1 </Customer>
<Restaurant> MC </Restaurant>
<Drink>
   <Drink> C1 </Drink>
   <Drink> C6 </Drink>
</Drink>
<Dish> D1 </Dish>
<Dish> D2 </Dish>
<Dish> D3 </Dish>
</Example>

我目前尝试的是：

import csv
import lxml.etree
from lxml import etree as ET
from lxml.builder import E

with open(r'restaurant.csv') as csvfile:
     result = E.Example(*
                         ( E.Customer(row['customer']),
                           E.Restaurant([row'restaurant']),
                           E.Drink(row['drink']),
                           E.Dish(row['dish']))

     for row in csv.DictReader(csv))
)

tree = lxml.etree.tostring(results, pretty_print=True )

with open(r'output.xml', 'wb') as f:
    f.write(tree)

我不知道如何在另一个循环中循环这个列表。

csv

来源：https://stackoverflow.com/questions/75397711/lxml-create-multiple-tags-while-looping-through-a-list-in-csv

1条答案

按热度按时间

cvxl0en21#

您可以尝试：

import csv
import lxml.etree
from lxml import etree as ET
from lxml.builder import E

with open('data.csv', 'r') as csvfile:
    reader = csv.DictReader(csvfile)

    example = E.Example()

    for row in reader:
        example.append(E.Customer(row['customer']))
        example.append(E.Restaurant(row['restaurant']))
        drink = E.Drink()
        example.append(drink)
        for d in map(str.strip, row['drink'].split(',')):
            drink.append(E.Drink(d))
        for d in map(str.strip, row['dish'].split(',')):
            example.append(E.Dish(d))

tree = lxml.etree.tostring(example, pretty_print=True)
print(tree.decode('utf-8'))

图纸：

<Example>
  <Customer>1</Customer>
  <Restaurant>MC</Restaurant>
  <Drink>
    <Drink>C1</Drink>
    <Drink>C6</Drink>
  </Drink>
  <Dish>D1</Dish>
  <Dish>D2</Dish>
  <Dish>D3</Dish>
</Example>

data.csv的含量：

customer,restaurant,drink,dish
1,MC,"C1, C6","D1, D2, D3"

赞(0）回复(0）举报 2023-02-10

我来回答

lxml：在csv格式的列表中循环时创建多个标记

1条答案

相关问题

热门标签

最新问答