我的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
我的问题有两个:
答案 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'