使用pandas python附加带字典值的csv

时间:2017-05-03 09:38:12

标签: python csv pandas dictionary

我的python脚本生成一个字典如下:

=============================================== =================

TL&安培; DR

我使用from_dict方法过度复杂化问题,同时从字典创建数据框。感谢@Sword。

换句话说,只有当您想要创建一个包含一列中所有键的数据帧时,才需要pd.DataFrame.from_dict,所有值都在另一列中。在所有其他情况下,它就像接受的答案中提到的方法一样简单。

=============================================== ===============

{u'19:00': 2, u'12:00': 1, u'06:00': 2, u'00:00': 0, u'23:00': 2, u'05:00': 2, u'11:00': 4, u'14:00': 2, u'04:00': 0, u'09:00': 7, u'03:00': 1, u'18:00': 6, u'01:00': 0, u'21:00': 5, u'15:00': 8, u'22:00': 1, u'08:00': 5, u'16:00': 8, u'02:00': 0, u'13:00': 8, u'20:00': 5, u'07:00': 11, u'17:00': 12, u'10:00': 8}

并且它还生成一个变量,假设full_name(作为脚本的参数),其值为“John”。

每次我运行脚本时,它都会以上述格式给我一个字典和名称。

我想将其写入csv文件,以便以后的格式进行分析:

FULLNAME | 00:00  |  01:00  |  02:00  | .....| 22:00  |  23:00  |
John     | 0      |  0      |  0      | .....| 1      |  2      |

我生成的代码如下:

import collections
import pandas as pd

# ........................
# Other part of code, which produces the dictionary by name "data_dict"
# ........................

#Sorting the dictionary (And adding it to a ordereddict) in order to skip matching dictionary keys with column headers
data_dict_sorted = collections.OrderedDict(sorted(data_dict.items()))

# For the first time to produce column headers, I used .items() and rest of the following lines follows it.
# df = pd.DataFrame.from_dict(data_dict_sorted.items())

#For the second time onwards, I just need to append the values, I am using .values()
df = pd.DataFrame.from_dict(data_dict_sorted.values())

df2 = df.T # transposing because from_dict creates all keys in one column, and corresponding values in the next column.
df2.columns = df2.iloc[0] 
df3 = df2[1:]
df3["FULLNAME"] = args.name #This is how we add a value, isn't it?
df3.to_csv('test.csv', mode = 'a', sep=str('\t'), encoding='utf-8', index=False)

我的代码正在生成以下csv

00:00 | 01:00 | 02:00 | …….. | 22:00 | 23:00 | FULLNAME
0     | 0     | 0     | …….. | 1     | 2     | John
0     | 0     | 0     | …….. | 1     | 2     | FULLNAME
0     | 0     | 0     | …….. | 1     | 2     | FULLNAME

我的问题有两个:

  1. 为什么在第二次迭代中打印“FULLNAME”而不是“John”(就像第二次运行脚本一样)?我缺少什么?
  2. 有更好的方法吗?

1 个答案:

答案 0 :(得分:1)

这个怎么样?

df = pd.DataFrame(data_dict, index=[0])
df['FullName'] = 'John'

编辑:
理解您执行操作的方式有点困难,但问题似乎是行df.columns = df.iloc[0]。我上面提到的代码不需要分配列名或转置操作。如果要在每次迭代时添加字典,请尝试:

data_dict['FullName'] = 'John'
df = df.append(pd.DataFrame(data_dict, index =[0]), ignore_index = True).reset_index()

如果每一行可能有不同的名称,那么df['FullName'] = 'John'将导致整个列等同于John。因此,作为更好的步骤,在您的dict中创建一个名为'FullName'的键,并使用适当的名称作为其值,以避免为整个列分配统一值,即

data_dict['FullName'] = 'John'