python解析pysparkDataframe中的xml列

zxlwwiss  于 2021-05-27  发布在  Spark
关注(0)|答案(1)|浏览(382)

我对pyspark比较陌生,正在尝试解决一个数据问题。我有一个pyspark df,它是用从mssqlserver提取的数据创建的,有两列:id(integer)和xmlmsg(string)。第二列xmlmsg包含xml格式的数据。目标是解析xmlmsg列,并使用从xml中提取的列在同一df中创建其他列。
以下是pyspark df的示例结构:

ID  XMLMsg
101 ...<a><b>name1</b><c>loc1</c></a>...<d>dept1</d>...
102 ...<a><b>name2</b><c>loc2</c></a>...<d>dept2</d>...
103 ...<a><b>name3</b><c>loc3</c></a>...<d>dept3</d>...

预期输出为:

ID  XMLMsg                                              b       c       d
101 ...<a><b>name1</b><c>loc1</c></a>...<d>dept1</d>... name1   loc1    dept1
102 ...<a><b>name2</b><c>loc2</c></a>...<d>dept2</d>... name2   loc2    dept2
103 ...<a><b>name3</b><c>loc3</c></a>...<d>dept3</d>... name3   loc3    dept3

我尝试了一些建议,根据我的搜索,所以;但未能达到预期效果。因此,寻求一些帮助和指导。谢谢你的时间。

fkaflof6

fkaflof61#

考虑到我必须从一个巨大的xml文件中获取4个节点的文本,我最终使用lambda和udf解决了这个问题。由于xml文件已经在一列中,并且是pysparkDataframe的一部分,所以我不想以文件的形式编写并再次解析整个xml。我还想避免使用xsd模式。实际的xml有多个名称空间和一些具有特定条件的节点。例子:

<ap:applicationproduct xmlns:xsi="http://www.example.com/2005/XMLSchema-instance" xmlns:ap="http://example.com/productinfo/1_6" xmlns:ct="http://example.com/commontypes/1_0" xmlns:dc="http://example.com/datacontent/1_0" xmlns:tp="http://aexample.com/prmvalue/1_0" ....." schemaVersion="..">
  <ap:ParameterInfo>
    <ap:Header>
      <ct:Version>1.0</ct:Version>
      <ct:Sender>ABC</ct:Sender>
      <ct:SenderVersion />
      <ct:SendTime>...</ct:SendTime>
    </ap:Header>
    <ap:ProductID>
      <ct:Model>
        <ct:Series>34AP</ct:Series>
        <ct:ModelNo>013780</ct:ModelNo>
     ..............
      ..............
   <ap:Object>
        <ap:Parameter schemaVersion="2.5" Code="DDA">
          <dc:Value>
            <tp:Blob>mbQAEAgBTgKQEBAX4KJJYABAIASL0AA==</tp:Blob>
          </dc:Value>
        </ap:Parameter>
.........
........

这里我需要从ct:modelno和tp:blob中提取值

from pyspark.sql.types import *
from pyspark.sql.functions import udf
import xml.etree.ElementTree as ET

# List of namespaces to be used:

ns = {'ap' : 'http://example.com/productinfo/1_6', 
'ct':'http://example.com/commontypes/1_0', 
'dc':'http://example.com/datacontent/1_0', 
'tp':'http://aexample.com/prmvalue/1_0' 
    }

parsed_model = (lambda x:    ET.fromstring(x).find('ap:ParameterInfo/ap:ProductID/ct:Model/ct:ModelNo', ns).text)
udf_model = udf(parsed_model)
parsed_model_df = df.withColumn('ModelNo', udf_Model('XMLMsg'))

对于具有blob值的节点,也可以编写类似的函数,但节点的路径为:“ap:parameterinfo/ap:object/ap:parameter[@code=“dda”]/dc:value/tp:blob”
这对我很有用,我能够在pyspark df中添加所需的值。不过,我们欢迎您提出任何建议,使之更好。谢谢您!

相关问题