将数据帧拆分为几列,Python



由于一些操作,我得到了一个看起来像这样的数据帧:

                0                                    1
0      (aut, aut)                           (1.0, 0.0)
1      (isr, pcn)   (0.0621031946211, 0.0840317734128)
2      (wlf, gum)   (0.00971778368827, 0.787082275372)
3      (lka, are)  (0.184325574632, 2.37291167033e-07)
4      (mmr, brb)  (-0.00659784629805, 0.854498462056)
5      (umi, mar)  (0.136002437743, 0.000146047773528)
6      (rwa, arm)  (0.143873473167, 5.82638804266e-05)

但是我需要将这个数据帧拆分为如下所示的内容:

      iso_a  iso_b     value_1               value_2
0      aut    aut        1.0                   0.0
1      isr    pcn   0.062103194621     0.0840317734128
2      wlf    gum   0.009717783688     0.787082275372
3      lka    are   0.184325574632     2.37291167033e-07
4      mmr    brb  -0.006597846298     0.854498462056
5      umi    mar   0.136002437743     0.000146047773528
6      rwa    arm   0.143873473167     5.82638804266e-05

我可能会:

def x(col):
    return col[0]
df['ios_a'] = df[0].apply(x)
df['value_1'] = df[1].apply(x)

def y(col):
    return col[1]
df['ios_b'] = df[0].apply(y)
df['value_2'] = df[1].apply(y)

然后,如果您愿意,可以删除前两列。

del df[0]
del df[1]

这有点笨拙(不是 DRY(,但可以完成工作。 def x(): 获取列(列 df[0]df[1] (,然后返回每行元组的第一部分,将其放入新分配的列中(例如 df['iso_a'] (然后def y():执行相同的操作,但这次返回每个元组的第二部分。这有意义吗?此外,这是假设您使用的是 Pandas 数据帧。

由于您提供了非常(没有(关于需要读取输入数据的格式的详细信息,因此这里有一种基本但简单的方法:

ls = []
with open('del.txt', 'r') as f:
    for line in f:
        ls.append(line.replace('(', '').replace(')', '').replace(',', '').split())

for l in ls[1:]:
    print(l)

这给出了一个列表,其中每行都有一个子列表,每个元素都存储为字符串:

['0', 'aut', 'aut', '1.0', '0.0']
['1', 'isr', 'pcn', '0.0621031946211', '0.0840317734128']
['2', 'wlf', 'gum', '0.00971778368827', '0.787082275372']
['3', 'lka', 'are', '0.184325574632', '2.37291167033e-07']
['4', 'mmr', 'brb', '-0.00659784629805', '0.854498462056']
['5', 'umi', 'mar', '0.136002437743', '0.000146047773528']
['6', 'rwa', 'arm', '0.143873473167', '5.82638804266e-05']

这是使用翻译方法的另一种方法,该方法会产生相同的结果

ls = []
with open('del.txt', 'r') as f:
    for line in f:
        ls.append(line.translate(None, "(),").split())

我不确定这是输入文件还是多维数组。假设您的输入数据框是一个多维数组,其中每个元素都有另一个包含两个元素的数组。

def getListOfDictionaries(dataFrame):
  newList = list()
  for row in dataFrame:
    newList.append({'iso_a': row[0][0],
                    'iso_b': row[0][1],
                    'value_1': row[1][0],
                    'value_2': row[1][1]})
  return newList

正如我所说,我不知道我们可以以什么格式期待输入数据

最新更新