从SAS数据集获取数据到Lua表

转身未来

1. data one;
   
2.   array c{2} $ _temporary_ ("A", "B");
   
3.   do i = 1 to dim(c);
   
4.     x = c[i];
   
5.     do j = 1 to 2;
   
6.       y = round(rannor(1), 0.0001);
   
7.       output;
   
8.     end;
   
9.   end;
   
10. run;
    
11. 
    
12. proc lua;
    
13. submit;
    
14.   -- OPEN SAS DATASET FOR READING --
    
15.   local dsid = sas.open("work.one", i)
    
16. 
    
17.   -- CREATING AN EMPTY LUA TABLE --
    
18.   local list = {}
    
19. 
    
20.   -- LOOP THROUGH OBSERVATIONS IN SAS DATASET --
    
21.   for obs in sas.rows(dsid) do
    
22.     local dict = {}
    
23. 
    
24.     -- LOOP THROUGH VARIABLES IN EACH OBSERVATION --
    
25.     for var in sas.vars(dsid) do
    
26.       dict[var.name] = obs[var.name]
    
27.     end
    
28. 
    
29.     -- INSERT EACH RECORD INTO LUA TABLE --
    
30.     table.insert(list, dict)
    
31. 
    
32.     -- CLOSE SAS DATASET AFTER THE LAST RECORD --
    
33.     if #list == sas.nobs(dsid) then
    
34.       sas.close(dsid)
    
35.     end
    
36.   end
    
37. 
    
38.   -- PRINT OUT LUA TABLE --
    
39.   for i = 1, #list do
    
40.     print(string.rep("*", 5).." RECORD: "..i.." "..string.rep("*", 5))
    
41.     for key, value in pairs(list[i]) do
    
42.       print(key.." --> "..type(value).." --> "..value)
    
43.     end
    
44.     print("\n")
    
45.   end
    
46.   -- WRITE LUA TABLE INTO NEW SAS DATASET --
    
47.   new_ds = "work.two"
    
48.   sas.write_ds(list, new_ds)
    
49. 
    
50.   -- SUBMITTING SAS CODE --
    
51.   sas.submit([[proc print data = @ds@ noobs; run]], {ds = new_ds})
    
52. 
    
53. endsubmit;
    
54. run;
    
55. 
    
56. *** OUTPUT SHOWN IN THE LOG ***
    
57. ***** RECORD: 1 *****
    
58. y --> number --> 1.8048
    
59. j --> number --> 1
    
60. i --> number --> 1
    
61. x --> string --> A
    
62. 
    
63. 
    
64. ***** RECORD: 2 *****
    
65. y --> number --> -0.0799
    
66. j --> number --> 2
    
67. i --> number --> 1
    
68. x --> string --> A
    
69. 
    
70. 
    
71. ***** RECORD: 3 *****
    
72. y --> number --> 0.3966
    
73. j --> number --> 1
    
74. i --> number --> 2
    
75. x --> string --> B
    
76. 
    
77. 
    
78. ***** RECORD: 4 *****
    
79. y --> number --> -1.0833
    
80. j --> number --> 2
    
81. i --> number --> 2
    
82. x --> string --> B
    

复制代码