Làm thế nào để bảo quản Nhãn khi tệp SPSS (.sav) được nhập vào gấu trúc qua rpy?

Question

Tôi đang tìm cách làm việc trên tệp SPSS (.sav) bằng cách sử dụng pandas. Trong sự vắng mặt của chương trình SPSS, đây là những gì một file điển hình trông giống như khi chuyển đổi sang .csv:

Mở điều tra về những gì hai hàng đầu tiên biểu thị (Tôi không biết SPSS), có vẻ như hàng đầu tiên chứa Label s, trong khi hàng thứ hai chứa VarName s.

Khi tôi đưa file vào gấu trúc như sau:

import pandas.rpy.common as com 

def savtocsv(filename): 
    w = com.robj.r('foreign::read.spss("%s", to.data.frame=TRUE)' % filename) 
    w = com.convert_robj(w) 
    return w 

và sau đó làm một cái đầu(), dòng đầu tiên (Label) là mất tích:

Nhãn có thể được duy trì như thế nào?

• Ref: Is there a Python module to open SPSS files?
• Python: 2.7.10
• Pandas: 0.17.1

Answer 1

Nhãn trong một file sav được lưu trữ trong variable.labels thuộc tính của đối tượng trở về từ read.spss chức năng.

Bạn có thể nhận được các nhãn biến như sau:

import pandas.rpy.common as com 

def get_labels(filename): 
    w = com.robj.r('attr(foreign::read.spss("%s"), "variable.labels")' % filename) 
    w = com.convert_robj(w) 
    return w 

Nếu bạn muốn thiết lập các nhãn như tên cột của dataframe của bạn:

import pandas.rpy.common as com 

def savtocsv(filename): 
    w = com.robj.r('foreign::read.spss("%s", to.data.frame=TRUE)' % filename) 
    cols = list(com.robj.r("attr")(w, "variable.labels")) 
    w = com.convert_robj(w) 
    w.columns = cols 
    return w