1

私が達成しようとしているのは、次のような csv ファイルです。

,h.r9,h.r8,h.r5,h.r4,h.r7,h.r6,h.r1,h.r3,h.r2,h.r10
chris,0,1,0,0,0,0,0,1,0,0
bob,0,0,1,1,1,1,0,0,0,1
dick,1,0,1,0,0,0,0,0,1,1
ab,1,2,0,0,0,0,0,0,0,0
emma,0,0,1,0,1,0,1,0,1,0

多少関係ありません:このようなマトリックスの実際の名前を知っている人はいますか? これにより、少なくともこの問題のグーグル検索が容易になります。

データはもともと次のような辞書にあります。

dict = {}
dict['ab'] = ["h.r1","h.r2"]
dict['bob'] = ["h.r3","h.r4","h.r5","h.r6","h.r10"]
dict['chris'] = ["h.r2","h.r8"]
dict['dick'] = ["h.r1","h.r10","h.r9","h.r3"]
dict['emma'] = ["h.r3","h.r5","h.r7","h.r9"]

したがって、人が辞書に変数を持っている場合、CSV には 1 があり、それ以外の場合はゼロになるはずです。

以下のコードは csv ファイルを作成しますが、すべてゼロです。1 と 0 の論理をどのように実装しますか? PHP/HTMLでこれを行うことができますが、Pythonに非常に慣れていないので、うまくいきません...また、この関数が既に組み込まれている可能性のあるライブラリはありますか? (たぶんうるさい?)

import csv

dict = {}
dict['ab'] = ["h.r1","h.r2"]
dict['bob'] = ["h.r3","h.r4","h.r5","h.r6","h.r10"]
dict['chris'] = ["h.r2","h.r8"]
dict['dick'] = ["h.r1","h.r10","h.r9","h.r3"]
dict['emma'] = ["h.r3","h.r5","h.r7","h.r9"]

mergedlist = []
for k,v in dict.items():
        mergedlist = mergedlist + v
mergedlist = list(set(mergedlist))
listlen = len(mergedlist)
mergedlist[0:0] = " "

thekeys = []
for k in dict.keys():
        thekeys.append(k)

thecsv = csv.writer(open("your.csv", 'w', newline=''))
thecsv.writerow(mergedlist)

for col in thekeys:
        thecsv.writerow([col] + ['0']*listlen) #does the logic go in here somehow??
4

3 に答える 3

3

そのコードのほとんどが何をしているのかわかりません。ほんの数行で実行できます。

all_values = set()
for person_values in data.values():
    all_values = all_values.union(person_values)
all_values_list = list(all_values)

thecsv = csv.writer(open("your.csv", 'w'))
thecsv.writerow(all_values_list)
for name, values in data.items():
    row = [1 if i in values else 0 for i in all_values_list]
    thecsv.writerow([name] + row)

data組み込みの関数/クラス名を上書きするのは良い考えではないため、辞書の名前を に変更したことに注意してください。

于 2013-08-24T21:44:04.197 に答える
1

私はそれを完全に異なる方法で行いますが、あなたのコードを使用して次のように変更します['0']*listlen:
['1' if x in dict[col] else '0' for x in thekeys]

于 2013-08-24T21:47:32.063 に答える
1

dictまず、オーバーライドするデータに名前を付けたくない場合は、次dictionaryのように変更しdict1ます。

#first generate the data body:
>>> ar=[[dict1[jtem].count(item) for item in ['h.r%s' %i for i in range(1, 11)]] for jtem in dict1] #are your values only in h.s1 to h.s10?
>>> ar
[[0, 1, 0, 0, 0, 0, 0, 1, 0, 0], \
[1, 0, 1, 0, 0, 0, 0, 0, 1, 1], \
[0, 0, 1, 1, 1, 1, 0, 0, 0, 1], \
[1, 1, 0, 0, 0, 0, 0, 0, 0, 0], \
[0, 0, 1, 0, 1, 0, 1, 0, 1, 0]]
#Then insert the names:
>>> junk=[item.insert(0, jtem) for item, jtem in zip(ar, dict1.keys())]
>>> ar
[['chris', 0, 1, 0, 0, 0, 0, 0, 1, 0, 0], \
['dick', 1, 0, 1, 0, 0, 0, 0, 0, 1, 1], \
['bob', 0, 0, 1, 1, 1, 1, 0, 0, 0, 1], \
['ab', 1, 1, 0, 0, 0, 0, 0, 0, 0, 0], \
['emma', 0, 0, 1, 0, 1, 0, 1, 0, 1, 0]]
#Then insert the title:
>>> ar.insert(0, ['',]+['h.r%s' %i for i in range(1, 11)])
>>> ar
[['', 'h.r1', 'h.r2', 'h.r3', 'h.r4', 'h.r5', 'h.r6', 'h.r7', 'h.r8', 'h.r9', 'h.r10'], \
['chris', 0, 1, 0, 0, 0, 0, 0, 1, 0, 0], \
['dick', 1, 0, 1, 0, 0, 0, 0, 0, 1, 1], \
['bob', 0, 0, 1, 1, 1, 1, 0, 0, 0, 1], \
['ab', 1, 1, 0, 0, 0, 0, 0, 0, 0, 0], \
['emma', 0, 0, 1, 0, 1, 0, 1, 0, 1, 0]]

次に、好みの方法で CSV ファイルに書き込みます。

于 2013-08-24T22:10:18.280 に答える