作为一个偶尔的脚本编写者,我已经在这个论坛上翻遍了所有的东西,到目前为止,它让我走了很远,但我被困在了寻找帮助的道路上。我正在尝试从CSV结构创建一个XML文档,目标是获得如下所示的内容:
ID,Type,Currency,Notional,Underlying,Maturity Date,Representation Type
ID1,COMMIT,EUR,100,,2018-06-01,Bond
ID2,COMMIT,AUD,110,,2018-03-25,Stock
并将其转换成这样。
<tradeRequests>
<tradeRequest>
<id>ID1</id>
<newDeals size="1">
<deal>
<id>ID1</id>
<terms>
<id>ID1</id>
<MaturityDate>2018-06-01</MaturityDate>
</terms>
</deal>
</newDeals>
</tradeRequest>
<tradeRequest>
<id>ID2</id>
<newDeals size="1">
<deal>
<id>ID2</id>
<terms>
<id>ID2</id>
<MaturityDate>2018-06-01</MaturityDate>
</terms>
</deal>
</newDeals>
</tradeRequest>
</tradeRequests>
问题是我的脚本似乎没有以正确的方式格式化项目,因为实际上每一行都应该是一个tradeRequest,但我看不到这种格式。
下面是我的代码片段,它将从数量大得多的列中提取列子集。
import csv
import xml.etree.ElementTree as ET
import xml.dom.minidom
tradeRequests = ET.Element("tradeRequests")
tradeRequest = ET.SubElement(tradeRequests, "tradeRequest")
newDeals = ET.SubElement(tradeRequest, "newDeals")
deal = ET.SubElement(newDeals, "deal")
dealid = ET.SubElement(deal, "id")
with open('TestCase.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
ET.SubElement(tradeRequest, "id").text = row['ID']
ET.SubElement(tradeRequest, "newDeals", {'size':"1"} )
ET.SubElement(dealid, "id").text = row['ID']
ET.SubElement(dealid, "maturityDate").text = row['Maturity Date']
tree = ET.ElementTree(tradeRequests)
tree.write("Testcase.xml" )
xml = xml.dom.minidom.parse('Testcase.xml')
pretty_xml_as_string = xml.toprettyxml()
print pretty_xml_as_string
问题是我似乎不能正确地嵌套这些项。我尝试过创建父/子组合,但没有成功。相反,基于该代码,我看到了如下所示的输出。
<tradeRequests>
<tradeRequest>
<newDeals>
<deal>
<id>
<id>ID1</id>
<maturityDate>2018-06-01</maturityDate>
<id>ID2</id>
<maturityDate>2018-03-25</maturityDate>
</id>
</deal>
</newDeals>
<id>ID1</id>
<newDeals size="1"/>
<id>ID2</id>
<newDeals size="1"/>
</tradeRequest>
</tradeRequests>
任何帮助都会一如既往的感谢。
我没有料到这个用例需要动态循环和创建元素
ID1,COMMIT,EUR,100,,2018-06-01,Bond
ID2,110,2018-03-25,Stock
ID2,110,2018-03-26,A
ID2,110,2018-03-26,B
ID2,110,2018-03-26,C
因此,实际上我需要创建一个元素,该元素将循环遍历ID2,并根据行数动态创建一个新元素,这是未知的。
因此,我的预期结果如下所示
<tradeRequests>
<ids>
<id>ID1</id>
<element>
<maturityDate>2018-06-01</maturityDate>
<type>Stock</type
<element>
</id>
<id>ID2</id>
<element>
<maturityDate>2018-03-25</maturityDate>
<type>A</type>
</element>
<element>
<maturityDate>2018-03-25</maturityDate>
<type>B</type>
</element>
<maturityDate>2018-03-25</maturityDate>
<type>C</type>
</element>
</id>
</tradeRequests>
发布于 2018-06-27 05:36:30
我强烈建议使用优秀的lxml
library.,它真的很快,因为它是一个基于C库libxml2的包装器,它包含了元素构建器object E
,这让你的工作变得非常容易:
import csv
import lxml.etree
from lxml.builder import E
with open('TestCase.csv') as csvfile:
results = E.tradeRequests(*(
E.tradeRequest(
E.id(row['ID']),
E.newDeals(
E.deal(
E.id(row['ID']),
E.terms(
E.id(row['ID']),
E.MaturityDate(row['Maturity Date']),
)
),
size="1",
)
) for row in csv.DictReader(csvfile))
)
print(lxml.etree.tostring(results, pretty_print=True))
结果:
<tradeRequests>
<tradeRequest>
<id>ID1</id>
<newDeals size="1">
<deal>
<id>ID1</id>
<terms>
<id>ID1</id>
<MaturityDate>2018-06-01</MaturityDate>
</terms>
</deal>
</newDeals>
</tradeRequest>
<tradeRequest>
<id>ID2</id>
<newDeals size="1">
<deal>
<id>ID2</id>
<terms>
<id>ID2</id>
<MaturityDate>2018-03-25</MaturityDate>
</terms>
</deal>
</newDeals>
</tradeRequest>
</tradeRequests>
https://stackoverflow.com/questions/51051566
复制相似问题