범주 데이터의 데이터 프레임을 사전으로 변환하는 방법

Question

JSON으로 변환해야하는 데이터 프레임이 있습니다. 우선 사전으로 바꾸는 것이 더 쉬울 것이라고 생각하지만, 어떻게해야 할지를 알 수 없습니다.

다음

NAME, CATEGORY, TAG 
Ex1, Education, Books 
Ex2, Transportation, Bus 
Ex3, Education, Schools 
Ex4, Education, Books 
Ex5, Markets, Stores 

내가 같이 할 데이터를 원하는 것입니다 : 내가 여기 js.d3

으로 시각화 할 수 있도록 내가 JSON으로 변환하는 데 필요한 데이터가 현재 모습입니다

Data = { 
    Education { 
     Books { 
      key: Ex1, 
      key: Ex2 
} 
Schools { 
key: Ex3 
} 
} 
Transportation { 
Bus { 
key: Ex2 
} 
} 
Markets { 
Stores { 
key: Ex5 
} 
} 

(내 JSON은 여기서는 완벽하지 않지만 일반적인 아이디어를 전달하고 싶다고 생각합니다.)

Answer 1

가장 안쪽 그룹 목록입니다, 그래서 당신은 한 번 이상 키로서 '키'를 사용할 수 없습니다

:

import json, string 

def to_json(file): 
    data = {} 
    for line in open(file): 
     fields = map(string.strip, line.split(',')) 
     categories = data.get(fields[1], {}) 
     tags = categories.get(fields[2], []) 
     tags.append(fields[0]) 
     categories[fields[2]] = tags 
     data[fields[1]] = categories 
    return json.dumps(data) 

print to_json('data.csv') 

결과 : 브렌트 Washburne의에

{"Markets": {"Stores": ["Ex5"]}, "Education": {"Schools": ["Ex3"], "Books": ["Ex1", "Ex4"]}, "Transportation": {"Bus": ["Ex2"]}} 

Answer 2

이 코드는 감사 위의 매우 유용한 답변. 지금은 태그 열을 제거해야했습니다. 왜냐하면 지금은 너무 복잡했기 때문입니다 (많은 행에는 쉼표로 구분 된 두 개 이상의 태그가있었습니다). 또한 이름에 연결된 원 (정수)을 추가했습니다. 여기에 있습니다 :

import json, string 
import pprint 

def to_json(file): 
data = {} 
for line in open(file): 
    fields = map(string.strip, line.split(',')) 
    categories = data.get(fields[1], []) 
    to_append = {}  
    to_append[fields[0]] = fields[3] 
    categories.append(to_append) 
    data[fields[1]] = categories 
return json.dumps(data) 

print to_json('data.csv')