Python使用pandasgroupby將CSV轉換為XML

我有一個需要使用 Python 轉換為 XML 的 csv。我是 Python 開發新手。

CSV 資料示例：

Amount,Code
CODE50,1246
CODE50,6290
CODE25,1077
CODE25,9790
CODE100,5319
CODE100,4988

必要的輸出 XML

<coupon-codes coupon-id="CODE50">
    <code>1246</code>
    <code>1246</code>
<coupon-codes/>
<coupon-codes coupon-id="CODE25">
    <code>1077</code>
    <code>9790</code>
<coupon-codes/>
<coupon-codes coupon-id="CODE100">
    <code>5319</code>
    <code>4988</code>
<coupon-codes/>

我的猜測是我必須使用 Pandas 將 csv 拉入，用于pandas.groupby對Amount列進行分組，然后將其推送到元素/子元素中以創建 xml，然后列印/推送到 xml 檔案。我無法groupby作業，也不知道如何將其推送到元素中，然后填充子元素。

uj5u.com熱心網友回復：

import xml.etree.ElementTree as ET
import pandas as pd

df = pd.read_csv('path/to/csvfile.csv')

root = ET.Element('root')
for coupon_id, df_group in df.groupby('Amount'):
    coupon_codes = ET.Element('coupon-codes', {'coupon-id': coupon_id})
    for coupon_code in df_group['Code']:
        code = ET.Element('code')
        code.text = str(coupon_code)
        coupon_codes.append(code)
    root.append(coupon_codes)

要將其轉換為縮小的字串，請使用以下命令：

min_xml = ET.tostring(root, encoding='utf8')
print(min_xml)

輸出：

<root><coupon-codes coupon-id="CODE100"><code>5319</code><code>4988</code></coupon-codes><coupon-codes coupon-id="CODE25"><code>1077</code><code>9790</code></coupon-codes><coupon-codes coupon-id="CODE50"><code>1246</code><code>6290</code></coupon-codes></root>

要將其轉換為格式化字串，請使用以下命令：

ET.indent(root)
pretty_xml = ET.tostring(root, encoding='utf8')
print(pretty_xml)

輸出：

<root>
  <coupon-codes coupon-id="CODE100">
    <code>5319</code>
    <code>4988</code>
  </coupon-codes>
  <coupon-codes coupon-id="CODE25">
    <code>1077</code>
    <code>9790</code>
  </coupon-codes>
  <coupon-codes coupon-id="CODE50">
    <code>1246</code>
    <code>6290</code>
  </coupon-codes>
</root>

uj5u.com熱心網友回復：

我還不能發表評論，但如果你把你用來解決問題的代碼放在那里會更好。

但是這里有一個非常相似的問題，simple-csv-to-xml-conversion-python

這是分組依據的帖子，分組依據/cumcount

有了這些資訊，我做到了，通過這樣做

df = pd.DataFrame(['CODE50',
    'CODE50',
    'CODE25',
    'CODE25',
    'CODE100',
    'CODE100'],columns=['code'])

df['amount'] = [1246,
    6290,
    1077,
    9790,
    5319,
    4988]

df['columns'] = df.groupby('code')['amount'].cumcount()
result = df.pivot(index='code', columns='columns')

def convert(df):
    str = ''
    for a in df.index:
        str  = '\n'f"""<coupon-codes coupon-id="{a}">"""
        for b in df['amount'].loc[a]:
            str  = '\n''       'f"<code>{b}</code>"
    return str
xml = convert(result)
print(xml)

結果

<coupon-codes coupon-id="CODE100">
       <code>5319</code>
       <code>4988</code>
<coupon-codes coupon-id="CODE25">
       <code>1077</code>
       <code>9790</code>
<coupon-codes coupon-id="CODE50">
       <code>1246</code>
       <code>6290</code>

uj5u.com熱心網友回復：

為什么不使用 df.to_xml()？這個簡單的 api 將從資料幀生成 xml

uj5u.com熱心網友回復：

ElementTree 可以毫不費力地做到這一點。見下文

from collections import defaultdict
from xml.etree.ElementTree import Element, SubElement,ElementTree

data = defaultdict(list)

with open('in.txt') as f:
  next(f)
  for line in f:
    parts = line.split(',')
    data[parts[0]].append(parts[1].strip())

root = Element('root')
for k,v in data.items():
  sub = SubElement(root,'coupon-codes',attrib={'coupon-id':k})
  for vv in v:
    sub_v = SubElement(sub,'code')
    sub_v.text = vv

tree = ElementTree(root)
with open('out.xml', 'w') as f:
  tree.write(f, encoding='unicode')

標籤：Python 熊猫 xml 文件 转型

上一篇：awk根據兩列和自定義重復規則洗掉重復項

下一篇：“utf-8”編解碼器無法解碼位置2912中的位元組0xd5：在Python中讀取csv檔案時出現無效的連續位元組錯誤