2012-02-29 421 views
2

創建CSV標題行變量我已經做了一些similar in Bash但不知道如何在Ruby中的2-3簡練的線條做到這一點,但現在看來,這可以用正確的編碼功夫瞬間完成-FU。如何在Ruby中

我有 「FILE.CSV」 有標題行,看起來像:

Ticker,"Price","Market Cap","Average Volume","Analyst Recom","Relative Strength Index (14)","Sector","Industry","Dividend Yield","Beta","52-Week Low","52-Week High","50-Day Low","50-Day High","Company","50-Day Simple Moving Average","Country","P/E","Forward P/E","PEG","P/S","P/B","P/Cash","P/Free Cash Flow","Payout Ratio","EPS (ttm)","EPS growth this year","EPS growth next year","EPS growth past 5 years","EPS growth next 5 years","Sales growth past 5 years","EPS growth quarter over quarter","Sales growth quarter over quarter","Shares Outstanding","Shares Float","Insider Ownership","Insider Transactions","Institutional Ownership","Institutional Transactions","Float Short","Short Ratio","Return on Assets","Return on Equity","Return on Investment","Current Ratio","Quick Ratio","LT Debt/Equity","Total Debt/Equity","Gross Margin","Operating Margin","Profit Margin","Performance (Week)","Performance (Month)","Performance (Quarter)","Performance (Half Year)","Performance (Year)","Performance (Year)","Average True Range","Volatility (Week)","Volatility (Month)","20-Day Simple Moving Average","200-Day Simple Moving Average","Change from Open","Gap","Relative Volume","Change","Volume","Earnings Date","No." 

,隨後爲約7000行看起來像:

FCD,27.89,,0.94,,66.75,"Financial","Exchange Traded Fund",3.13%,,19.75%,-0.36%,6.37%,-0.36%,"Focus Morningstar Consumer Defensive ETF",2.28%,"USA",,,,,,,,,,,,,,,, 
,,,,,,,,,,,,,,,,,,,0.36%,3.07%,9.93%,10.85%,,2.01%,0.12,0.04%,0.21%,1.26%,6.69%,0.00%,-0.04%,0.96,-0.04%,900,,2186 
FCE-A,14.59,2496.64,960.33,2.50,54.76,"Financial","Property Management",,2.83,56.55%,-24.87%,36.61%,-7.77%,"Forest City Enterprises Inc.",11.49%,"USA",,69.48,,2.2 
5,1.58,10.87,,,-0.02,410.77%,250.00%,-10.06%,8.00%,1.54%,-28.77%,-9.00%,171.12,136.94,0.26%,-8.25%,74.80%,-0.13%,4.62%,6.59,0.46%,-0.12%,0.54%,,,4.35,4.35,39.54%, 
4.82%,4.60%,-4.01%,8.96%,25.45%,13.10%,-22.80%,23.43%,0.44,3.07%,2.98%,-0.89%,1.49%,-1.62%,0.00%,0.47,-1.62%,449874,12/8/2010 4:30:00 PM,2187 

給定一個股票代碼爲 「FCD」我正在嘗試將大約30個從頭字段中取出的新變量集中分配給匹配「FCD」行的值。

每個新變量都以fv_作爲前綴,並且在字段名稱中減去所有標點符號,空格,引號等(變量不友好的東西)。

所以對於「FCD」我想給我的腳本:

fv_Ticker="FCD" 
fv_Price=27.89 
fv_MarketCap="" 
fv_VolatilityMonth=0.21 # if get String not Float because of trailing % in "0.21%" that's okay, will deal with it later 
etc. 

我要指出,我退出使用任何類型的CSV.readCSV.foreach由於horrible slowdown using native Ruby 1.9.x CSV objects,歷時分鐘閱讀,因此是不可接受的在重複運行的實時應用程序中。

相反,我一直在使用一個Ruby管「的awk」指定從文件瞬間這樣讀取各個變量:

$stock="FCD" 
$dividend_yield = IO.readlines("|awk -F, '$1==\"#{$stock}\" {print $9}' finviz.AllStocks.csv")[0].to_f 
$beta = IO.readlines("|awk -F, '$1==\"#{$stock}\" {print $10}' AllStocks.csv")[0].to_f 

,但現在它變得太毛茸茸到不能一概而論。它需要處理任何帶有未知字段的CSV文件,直到它看到的第一行。

+0

爲什麼變量?爲什麼不是'{「FCD」=> someObject}'的散列? – Linuxios 2012-02-29 16:09:20

+0

散列也可以,如果不是更好,但它需要很快。 [先前](http://stackoverflow.com/q/8476769/1069375)將CSV讀入Ruby失敗了。 – Marcos 2012-02-29 16:23:12

+0

用變量填充符號表不能比散列效率低。而且,對任何事情使用全局變量,尤其是這個,不是一個好主意。 – Linuxios 2012-02-29 16:30:10

回答

0

想想我正在做些什麼。中間的路線的方式,讓紅寶石只看到兩行CSV輸入,所以本地處理要快:

puts IO.readlines(%`|sed -n "1p; /^#{$stock},/p" AllStocks.csv`) # sed filters out the gazillion other lines that make Ruby slow 

它的工作原理。由於紅寶石僅看到一個短CSV,一條線的標頭和用於由於sed數據一行,它加載在一瞬間,我有我的散列:(關聯數組我之後)

$stock = "ANAD" 

csv_data = CSV.parse IO.read(%`|sed -n "1p; /^#{$stock},/p" AllStocks.csv`) 
headers = csv_data.shift.map {|i| i.to_s } 
string_data = csv_data.map {|row| row.map {|cell| cell.to_s } } 
tickers = array_of_hashes = string_data.map {|row| Hash[*headers.zip(row).flatten] } 

pp tickers[0]  
=> {"Ticker"=>"ANAD", "Price"=>"2.57", "Market Cap"=>"175.20", "Average Volume"=>"442.65", "Analyst Recom"=>"2.90", "Relative Strength Index (14)"=>"38.21", "Sector"=>"Technology", "Industry"=>"Semiconductor - Integrated Circuits", "Dividend Yield"=>"", "Beta"=>"2.30", "52-Week Low"=>"33.85%", "52-Week High"=>"-52.84%", "50-Day Low"=>"27.23%", "50-Day High"=>"-20.19%", "Company"=>"Anadigics, Inc.", "50-Day Simple Moving Average"=>"-2.63%", "Country"=>"USA", "P/E"=>"", "Forward P/E"=>"", "PEG"=>"", "P/S"=>"1.15", "P/B"=>"1.05", "P/Cash"=>"3.08", "P/Free Cash Flow"=>"", "Payout Ratio"=>"", "EPS (ttm)"=>"-0.73", "EPS growth this year"=>"-4002.36%", "EPS growth next year"=>"57.90%", "EPS growth past 5 years"=>"-7.14%", "EPS growth next 5 years"=>"15.67%", "Sales growth past 5 years"=>"-1.69%", "EPS growth quarter over quarter"=>"-564.96%", "Sales growth quarter over quarter"=>"-39.37%", "Shares Outstanding"=>"68.17", "Shares Float"=>"66.25", "Insider Ownership"=>"3.87%", "Insider Transactions"=>"-7.51%", "Institutional Ownership"=>"47.64%", "Institutional Transactions"=>"-6.24%", "Float Short"=>"3.30%", "Short Ratio"=>"4.95", "Return on Assets"=>"-23.34%", "Return on Equity"=>"-26.70%", "Return on Investment"=>"-26.33%", "Current Ratio"=>"4.89", "Quick Ratio"=>"3.90", "LT Debt/Equity"=>"0.00", "Total Debt/Equity"=>"0.00", "Gross Margin"=>"20.35%", "Operating Margin"=>"-32.79%", "Profit Margin"=>"-32.27%", "Performance (Week)"=>"-5.17%", "Performance (Month)"=>"-7.22%", "Performance (Quarter)"=>"21.23%", "Performance (Half Year)"=>"-6.20%", "Performance (Year)"=>"17.35%", "Average True Range"=>"0.16", "Volatility (Week)"=>"5.14%", "Volatility (Month)"=>"5.51%", "20-Day Simple Moving Average"=>"-13.04%", "200-Day Simple Moving Average"=>"-3.71%", "Change from Open"=>"-4.10%", "Gap"=>"0.00%", "Relative Volume"=>"0.95", "Change"=>"-4.10%", "Volume"=>"421200", "Earnings Date"=>"2/22/2012 7:00:00 AM", "No."=>"280"} 


tickers[0]["Volatility (Month)"].to_f 
=> 5.51 
1

這單行版甚至比last還要好。這是短,更簡單,有什麼特別吸引人,以適應所有的頭標題爲(almost-)可變準備唯一的名字,所以我不擔心以後解析出大部分的東西:

csv_data = CSV.parse IO.read(%`|sed -n "1p; /^#{$stock},/p" AllStocks.csv`), {:headers => true, :return_headers => false, :header_converters => :symbol, :converters => :all} 

結果:

csv_data.size 
=> 1 

csv_data.to_a 
=> [[:ticker, :price, :market_cap, :average_volume, :analyst_recom, :relative_strength_index_14, :sector, :industry, :dividend_yield, :beta, :"52week_low", :"52week_high", :"50day_low", :"50day_high", :company, :"50day_simple_moving_average", :country, :pe, :forward_pe, :peg, :ps, :pb, :pcash, :pfree_cash_flow, :payout_ratio, :eps_ttm, :eps_growth_this_year, :eps_growth_next_year, :eps_growth_past_5_years, :eps_growth_next_5_years, :sales_growth_past_5_years, :eps_growth_quarter_over_quarter, :sales_growth_quarter_over_quarter, :shares_outstanding, :shares_float, :insider_ownership, :insider_transactions, :institutional_ownership, :institutional_transactions, :float_short, :short_ratio, :return_on_assets, :return_on_equity, :return_on_investment, :current_ratio, :quick_ratio, :lt_debtequity, :total_debtequity, :gross_margin, :operating_margin, :profit_margin, :performance_week, :performance_month, :performance_quarter, :performance_half_year, :performance_year, :performance_year, :average_true_range, :volatility_week, :volatility_month, :"20day_simple_moving_average", :"200day_simple_moving_average", :change_from_open, :gap, :relative_volume, :change, :volume, :earnings_date, :no], ["ANAD", 2.57, 175.2, 442.65, 2.9, 38.21, "Technology", "Semiconductor - Integrated Circuits", nil, 2.3, "33.85%", "-52.84%", "27.23%", "-20.19%", "Anadigics, Inc.", "-2.63%", "USA", nil, nil, nil, 1.15, 1.05, 3.08, nil, nil, -0.73, "-4002.36%", "57.90%", "-7.14%", "15.67%", "-1.69%", "-564.96%", "-39.37%", 68.17, 66.25, "3.87%", "-7.51%", "47.64%", "-6.24%", "3.30%", 4.95, "-23.34%", "-26.70%", "-26.33%", 4.89, 3.9, 0.0, 0.0, "20.35%", "-32.79%", "-32.27%", "-5.17%", "-7.22%", "21.23%", "-6.20%", "-50.39%", "17.35%", 0.16, "5.14%", "5.51%", "-13.04%", "-3.71%", "-4.10%", "0.00%", 0.95, "-4.10%", 421200, "2/22/2012 7:00:00 AM", 280]] 


$company = csv_data[:company][0] 
=> "Anadigics, Inc." 

csv_data[:volatility_month] 
=> ["5.51%"]