pandas dataframe 由用户输入

Question

我有一个关于通过用户输入导入 dataframe 的问题。 我想通过用户输入创建一个 pandas dataframe 。 第一行包含数据中的行数。 第二行包含数据帧的逗号分隔标头。 第三行及后续行包含 dataframe 数据。

例如我有第 1 行：3 第 2 行：变量 1、变量 2、变量 3 第 3 行：3、4、8 第 4 行：5、6、9 第 5 行：1、7、4

你能帮我输入 function 吗？

Answer 1

像这样：

In [48]: total_rows = int(input("Enter number of data rows:"))                                                                                                                                              
Enter number of data rows:3

In [50]: header = input("Enter header:")                                                                                                                                                                    
Enter header:variable 1, variable 2, variable 3

In [51]: data = [] 

In [63]: i = 1                                                                                                                                                                                              

In [64]: while i <= total_rows: 
    ...:     row = input("Enter data:") 
    ...:     data.append(row) 
    ...:     i += 1 
    ...:                                                                                                                                                                                                    
Enter data:3, 4, 8
Enter data:5, 6, 9
Enter data:1, 7, 4

In [66]: header = header.split(',')
In [74]: data = [x.split(',') for x in data]                                                                                                                                                                

In [75]: df = pd.DataFrame(data, columns=header)                                                                                                                                                            

In [76]: df                                                                                                                                                                                                 
Out[76]: 
  variable 1  variable 2  variable 3
0          3           4           8
1          5           6           9
2          1           7           4