[英]Saving header from CSV file using `numpy.genfromtxt()`
我正在使用numpy.genfromtxt()
來讀取CSV文件,我想將標題與標題下面的數據分開保存。
我知道skip_header=1
參數允許我跳過標題,但是,在這種情況下,標題會丟失,但我想保存它。 我嘗試使用skip_footer
參數,以便它跳過標題下面的所有內容,並通過將skip_footer
設置為小於CSV文件的長度或skip_footer=(len('filename.csv')-1)
來保留標題。 代碼運行但它沒有給出正確的輸出。 不知何故, numpy.genfromtxt()
不會以我想象的方式計算CSV文件的行數。
header = numpy.genfromtxt('filename.csv', delimiter=',', skip_footer=(len('filename.csv')-1))
我希望只將標題作為一維NumPy數組,而是得到類似整個數組的東西:
[[ nan nan nan ... nan nan nan]
[2.016e+03 1.000e+00 1.000e+00 ... 1.165e+01 6.999e+01 1.000e+00]
[2.016e+03 1.000e+00 1.000e+00 ... 8.000e+00 5.430e+01 1.000e+00]
...
[2.016e+03 6.000e+00 3.000e+01 ... 0.000e+00 4.630e+01 2.000e+00]
[2.016e+03 6.000e+00 3.000e+01 ... 8.750e+00 5.255e+01 1.000e+00]
[2.016e+03 6.000e+00 3.000e+01 ... 8.880e+00 5.822e+01 1.000e+00]]
我想保留在nan
的最頂端的東西。
解:
header = np.genfromtxt('filename.csv', delimiter=',', dtype=str, max_rows=1)
print(header)
OUTPUT:
['pickup_year' 'pickup_month' 'pickup_day' 'pickup_dayofweek'
'pickup_time' 'pickup_location_code' 'dropoff_location_code'
'trip_distance' 'trip_length' 'fare_amount' 'fees_amount' 'tolls_amount'
'tip_amount' 'total_amount' 'payment_type']
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.