csv 1.0.2 → 3.2.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/NEWS.md +868 -0
- data/README.md +6 -3
- data/doc/csv/arguments/io.rdoc +5 -0
- data/doc/csv/options/common/col_sep.rdoc +57 -0
- data/doc/csv/options/common/quote_char.rdoc +42 -0
- data/doc/csv/options/common/row_sep.rdoc +91 -0
- data/doc/csv/options/generating/force_quotes.rdoc +17 -0
- data/doc/csv/options/generating/quote_empty.rdoc +12 -0
- data/doc/csv/options/generating/write_converters.rdoc +25 -0
- data/doc/csv/options/generating/write_empty_value.rdoc +15 -0
- data/doc/csv/options/generating/write_headers.rdoc +29 -0
- data/doc/csv/options/generating/write_nil_value.rdoc +14 -0
- data/doc/csv/options/parsing/converters.rdoc +46 -0
- data/doc/csv/options/parsing/empty_value.rdoc +13 -0
- data/doc/csv/options/parsing/field_size_limit.rdoc +39 -0
- data/doc/csv/options/parsing/header_converters.rdoc +43 -0
- data/doc/csv/options/parsing/headers.rdoc +63 -0
- data/doc/csv/options/parsing/liberal_parsing.rdoc +38 -0
- data/doc/csv/options/parsing/nil_value.rdoc +12 -0
- data/doc/csv/options/parsing/return_headers.rdoc +22 -0
- data/doc/csv/options/parsing/skip_blanks.rdoc +31 -0
- data/doc/csv/options/parsing/skip_lines.rdoc +37 -0
- data/doc/csv/options/parsing/strip.rdoc +15 -0
- data/doc/csv/options/parsing/unconverted_fields.rdoc +27 -0
- data/doc/csv/recipes/filtering.rdoc +158 -0
- data/doc/csv/recipes/generating.rdoc +298 -0
- data/doc/csv/recipes/parsing.rdoc +545 -0
- data/doc/csv/recipes/recipes.rdoc +6 -0
- data/lib/csv/core_ext/array.rb +1 -1
- data/lib/csv/core_ext/string.rb +1 -1
- data/lib/csv/fields_converter.rb +89 -0
- data/lib/csv/input_record_separator.rb +18 -0
- data/lib/csv/parser.rb +1290 -0
- data/lib/csv/row.rb +505 -136
- data/lib/csv/table.rb +791 -114
- data/lib/csv/version.rb +1 -1
- data/lib/csv/writer.rb +210 -0
- data/lib/csv.rb +2432 -1329
- metadata +66 -13
- data/news.md +0 -112
data/lib/csv/table.rb
CHANGED
@@ -3,28 +3,210 @@
|
|
3
3
|
require "forwardable"
|
4
4
|
|
5
5
|
class CSV
|
6
|
+
# = \CSV::Table
|
7
|
+
# A \CSV::Table instance represents \CSV data.
|
8
|
+
# (see {class CSV}[../CSV.html]).
|
6
9
|
#
|
7
|
-
#
|
8
|
-
#
|
9
|
-
#
|
10
|
+
# The instance may have:
|
11
|
+
# - Rows: each is a Table::Row object.
|
12
|
+
# - Headers: names for the columns.
|
10
13
|
#
|
11
|
-
#
|
12
|
-
# row processing is activated.
|
14
|
+
# === Instance Methods
|
13
15
|
#
|
16
|
+
# \CSV::Table has three groups of instance methods:
|
17
|
+
# - Its own internally defined instance methods.
|
18
|
+
# - Methods included by module Enumerable.
|
19
|
+
# - Methods delegated to class Array.:
|
20
|
+
# * Array#empty?
|
21
|
+
# * Array#length
|
22
|
+
# * Array#size
|
23
|
+
#
|
24
|
+
# == Creating a \CSV::Table Instance
|
25
|
+
#
|
26
|
+
# Commonly, a new \CSV::Table instance is created by parsing \CSV source
|
27
|
+
# using headers:
|
28
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
29
|
+
# table = CSV.parse(source, headers: true)
|
30
|
+
# table.class # => CSV::Table
|
31
|
+
#
|
32
|
+
# You can also create an instance directly. See ::new.
|
33
|
+
#
|
34
|
+
# == Headers
|
35
|
+
#
|
36
|
+
# If a table has headers, the headers serve as labels for the columns of data.
|
37
|
+
# Each header serves as the label for its column.
|
38
|
+
#
|
39
|
+
# The headers for a \CSV::Table object are stored as an \Array of Strings.
|
40
|
+
#
|
41
|
+
# Commonly, headers are defined in the first row of \CSV source:
|
42
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
43
|
+
# table = CSV.parse(source, headers: true)
|
44
|
+
# table.headers # => ["Name", "Value"]
|
45
|
+
#
|
46
|
+
# If no headers are defined, the \Array is empty:
|
47
|
+
# table = CSV::Table.new([])
|
48
|
+
# table.headers # => []
|
49
|
+
#
|
50
|
+
# == Access Modes
|
51
|
+
#
|
52
|
+
# \CSV::Table provides three modes for accessing table data:
|
53
|
+
# - \Row mode.
|
54
|
+
# - Column mode.
|
55
|
+
# - Mixed mode (the default for a new table).
|
56
|
+
#
|
57
|
+
# The access mode for a\CSV::Table instance affects the behavior
|
58
|
+
# of some of its instance methods:
|
59
|
+
# - #[]
|
60
|
+
# - #[]=
|
61
|
+
# - #delete
|
62
|
+
# - #delete_if
|
63
|
+
# - #each
|
64
|
+
# - #values_at
|
65
|
+
#
|
66
|
+
# === \Row Mode
|
67
|
+
#
|
68
|
+
# Set a table to row mode with method #by_row!:
|
69
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
70
|
+
# table = CSV.parse(source, headers: true)
|
71
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
72
|
+
#
|
73
|
+
# Specify a single row by an \Integer index:
|
74
|
+
# # Get a row.
|
75
|
+
# table[1] # => #<CSV::Row "Name":"bar" "Value":"1">
|
76
|
+
# # Set a row, then get it.
|
77
|
+
# table[1] = CSV::Row.new(['Name', 'Value'], ['bam', 3])
|
78
|
+
# table[1] # => #<CSV::Row "Name":"bam" "Value":3>
|
79
|
+
#
|
80
|
+
# Specify a sequence of rows by a \Range:
|
81
|
+
# # Get rows.
|
82
|
+
# table[1..2] # => [#<CSV::Row "Name":"bam" "Value":3>, #<CSV::Row "Name":"baz" "Value":"2">]
|
83
|
+
# # Set rows, then get them.
|
84
|
+
# table[1..2] = [
|
85
|
+
# CSV::Row.new(['Name', 'Value'], ['bat', 4]),
|
86
|
+
# CSV::Row.new(['Name', 'Value'], ['bad', 5]),
|
87
|
+
# ]
|
88
|
+
# table[1..2] # => [["Name", #<CSV::Row "Name":"bat" "Value":4>], ["Value", #<CSV::Row "Name":"bad" "Value":5>]]
|
89
|
+
#
|
90
|
+
# === Column Mode
|
91
|
+
#
|
92
|
+
# Set a table to column mode with method #by_col!:
|
93
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
94
|
+
# table = CSV.parse(source, headers: true)
|
95
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
96
|
+
#
|
97
|
+
# Specify a column by an \Integer index:
|
98
|
+
# # Get a column.
|
99
|
+
# table[0]
|
100
|
+
# # Set a column, then get it.
|
101
|
+
# table[0] = ['FOO', 'BAR', 'BAZ']
|
102
|
+
# table[0] # => ["FOO", "BAR", "BAZ"]
|
103
|
+
#
|
104
|
+
# Specify a column by its \String header:
|
105
|
+
# # Get a column.
|
106
|
+
# table['Name'] # => ["FOO", "BAR", "BAZ"]
|
107
|
+
# # Set a column, then get it.
|
108
|
+
# table['Name'] = ['Foo', 'Bar', 'Baz']
|
109
|
+
# table['Name'] # => ["Foo", "Bar", "Baz"]
|
110
|
+
#
|
111
|
+
# === Mixed Mode
|
112
|
+
#
|
113
|
+
# In mixed mode, you can refer to either rows or columns:
|
114
|
+
# - An \Integer index refers to a row.
|
115
|
+
# - A \Range index refers to multiple rows.
|
116
|
+
# - A \String index refers to a column.
|
117
|
+
#
|
118
|
+
# Set a table to mixed mode with method #by_col_or_row!:
|
119
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
120
|
+
# table = CSV.parse(source, headers: true)
|
121
|
+
# table.by_col_or_row! # => #<CSV::Table mode:col_or_row row_count:4>
|
122
|
+
#
|
123
|
+
# Specify a single row by an \Integer index:
|
124
|
+
# # Get a row.
|
125
|
+
# table[1] # => #<CSV::Row "Name":"bar" "Value":"1">
|
126
|
+
# # Set a row, then get it.
|
127
|
+
# table[1] = CSV::Row.new(['Name', 'Value'], ['bam', 3])
|
128
|
+
# table[1] # => #<CSV::Row "Name":"bam" "Value":3>
|
129
|
+
#
|
130
|
+
# Specify a sequence of rows by a \Range:
|
131
|
+
# # Get rows.
|
132
|
+
# table[1..2] # => [#<CSV::Row "Name":"bam" "Value":3>, #<CSV::Row "Name":"baz" "Value":"2">]
|
133
|
+
# # Set rows, then get them.
|
134
|
+
# table[1] = CSV::Row.new(['Name', 'Value'], ['bat', 4])
|
135
|
+
# table[2] = CSV::Row.new(['Name', 'Value'], ['bad', 5])
|
136
|
+
# table[1..2] # => [["Name", #<CSV::Row "Name":"bat" "Value":4>], ["Value", #<CSV::Row "Name":"bad" "Value":5>]]
|
137
|
+
#
|
138
|
+
# Specify a column by its \String header:
|
139
|
+
# # Get a column.
|
140
|
+
# table['Name'] # => ["foo", "bat", "bad"]
|
141
|
+
# # Set a column, then get it.
|
142
|
+
# table['Name'] = ['Foo', 'Bar', 'Baz']
|
143
|
+
# table['Name'] # => ["Foo", "Bar", "Baz"]
|
14
144
|
class Table
|
145
|
+
# :call-seq:
|
146
|
+
# CSV::Table.new(array_of_rows, headers = nil) -> csv_table
|
147
|
+
#
|
148
|
+
# Returns a new \CSV::Table object.
|
149
|
+
#
|
150
|
+
# - Argument +array_of_rows+ must be an \Array of CSV::Row objects.
|
151
|
+
# - Argument +headers+, if given, may be an \Array of Strings.
|
152
|
+
#
|
153
|
+
# ---
|
154
|
+
#
|
155
|
+
# Create an empty \CSV::Table object:
|
156
|
+
# table = CSV::Table.new([])
|
157
|
+
# table # => #<CSV::Table mode:col_or_row row_count:1>
|
158
|
+
#
|
159
|
+
# Create a non-empty \CSV::Table object:
|
160
|
+
# rows = [
|
161
|
+
# CSV::Row.new([], []),
|
162
|
+
# CSV::Row.new([], []),
|
163
|
+
# CSV::Row.new([], []),
|
164
|
+
# ]
|
165
|
+
# table = CSV::Table.new(rows)
|
166
|
+
# table # => #<CSV::Table mode:col_or_row row_count:4>
|
167
|
+
#
|
168
|
+
# ---
|
15
169
|
#
|
16
|
-
#
|
17
|
-
#
|
170
|
+
# If argument +headers+ is an \Array of Strings,
|
171
|
+
# those Strings become the table's headers:
|
172
|
+
# table = CSV::Table.new([], headers: ['Name', 'Age'])
|
173
|
+
# table.headers # => ["Name", "Age"]
|
18
174
|
#
|
19
|
-
#
|
20
|
-
#
|
175
|
+
# If argument +headers+ is not given and the table has rows,
|
176
|
+
# the headers are taken from the first row:
|
177
|
+
# rows = [
|
178
|
+
# CSV::Row.new(['Foo', 'Bar'], []),
|
179
|
+
# CSV::Row.new(['foo', 'bar'], []),
|
180
|
+
# CSV::Row.new(['FOO', 'BAR'], []),
|
181
|
+
# ]
|
182
|
+
# table = CSV::Table.new(rows)
|
183
|
+
# table.headers # => ["Foo", "Bar"]
|
21
184
|
#
|
22
|
-
#
|
23
|
-
#
|
24
|
-
#
|
185
|
+
# If argument +headers+ is not given and the table is empty (has no rows),
|
186
|
+
# the headers are also empty:
|
187
|
+
# table = CSV::Table.new([])
|
188
|
+
# table.headers # => []
|
25
189
|
#
|
26
|
-
|
190
|
+
# ---
|
191
|
+
#
|
192
|
+
# Raises an exception if argument +array_of_rows+ is not an \Array object:
|
193
|
+
# # Raises NoMethodError (undefined method `first' for :foo:Symbol):
|
194
|
+
# CSV::Table.new(:foo)
|
195
|
+
#
|
196
|
+
# Raises an exception if an element of +array_of_rows+ is not a \CSV::Table object:
|
197
|
+
# # Raises NoMethodError (undefined method `headers' for :foo:Symbol):
|
198
|
+
# CSV::Table.new([:foo])
|
199
|
+
def initialize(array_of_rows, headers: nil)
|
27
200
|
@table = array_of_rows
|
201
|
+
@headers = headers
|
202
|
+
unless @headers
|
203
|
+
if @table.empty?
|
204
|
+
@headers = []
|
205
|
+
else
|
206
|
+
@headers = @table.first.headers
|
207
|
+
end
|
208
|
+
end
|
209
|
+
|
28
210
|
@mode = :col_or_row
|
29
211
|
end
|
30
212
|
|
@@ -40,103 +222,295 @@ class CSV
|
|
40
222
|
extend Forwardable
|
41
223
|
def_delegators :@table, :empty?, :length, :size
|
42
224
|
|
225
|
+
# :call-seq:
|
226
|
+
# table.by_col -> table_dup
|
43
227
|
#
|
44
|
-
# Returns a duplicate
|
45
|
-
#
|
46
|
-
#
|
228
|
+
# Returns a duplicate of +self+, in column mode
|
229
|
+
# (see {Column Mode}[#class-CSV::Table-label-Column+Mode]):
|
230
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
231
|
+
# table = CSV.parse(source, headers: true)
|
232
|
+
# table.mode # => :col_or_row
|
233
|
+
# dup_table = table.by_col
|
234
|
+
# dup_table.mode # => :col
|
235
|
+
# dup_table.equal?(table) # => false # It's a dup
|
47
236
|
#
|
48
|
-
# This
|
49
|
-
#
|
50
|
-
#
|
237
|
+
# This may be used to chain method calls without changing the mode
|
238
|
+
# (but also will affect performance and memory usage):
|
239
|
+
# dup_table.by_col['Name']
|
51
240
|
#
|
241
|
+
# Also note that changes to the duplicate table will not affect the original.
|
52
242
|
def by_col
|
53
243
|
self.class.new(@table.dup).by_col!
|
54
244
|
end
|
55
245
|
|
246
|
+
# :call-seq:
|
247
|
+
# table.by_col! -> self
|
56
248
|
#
|
57
|
-
#
|
58
|
-
#
|
59
|
-
#
|
60
|
-
#
|
61
|
-
#
|
249
|
+
# Sets the mode for +self+ to column mode
|
250
|
+
# (see {Column Mode}[#class-CSV::Table-label-Column+Mode]); returns +self+:
|
251
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
252
|
+
# table = CSV.parse(source, headers: true)
|
253
|
+
# table.mode # => :col_or_row
|
254
|
+
# table1 = table.by_col!
|
255
|
+
# table.mode # => :col
|
256
|
+
# table1.equal?(table) # => true # Returned self
|
62
257
|
def by_col!
|
63
258
|
@mode = :col
|
64
259
|
|
65
260
|
self
|
66
261
|
end
|
67
262
|
|
263
|
+
# :call-seq:
|
264
|
+
# table.by_col_or_row -> table_dup
|
68
265
|
#
|
69
|
-
# Returns a duplicate
|
70
|
-
#
|
71
|
-
#
|
266
|
+
# Returns a duplicate of +self+, in mixed mode
|
267
|
+
# (see {Mixed Mode}[#class-CSV::Table-label-Mixed+Mode]):
|
268
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
269
|
+
# table = CSV.parse(source, headers: true).by_col!
|
270
|
+
# table.mode # => :col
|
271
|
+
# dup_table = table.by_col_or_row
|
272
|
+
# dup_table.mode # => :col_or_row
|
273
|
+
# dup_table.equal?(table) # => false # It's a dup
|
72
274
|
#
|
73
|
-
# This
|
74
|
-
#
|
75
|
-
#
|
275
|
+
# This may be used to chain method calls without changing the mode
|
276
|
+
# (but also will affect performance and memory usage):
|
277
|
+
# dup_table.by_col_or_row['Name']
|
76
278
|
#
|
279
|
+
# Also note that changes to the duplicate table will not affect the original.
|
77
280
|
def by_col_or_row
|
78
281
|
self.class.new(@table.dup).by_col_or_row!
|
79
282
|
end
|
80
283
|
|
284
|
+
# :call-seq:
|
285
|
+
# table.by_col_or_row! -> self
|
81
286
|
#
|
82
|
-
#
|
83
|
-
#
|
84
|
-
#
|
85
|
-
#
|
86
|
-
#
|
87
|
-
#
|
88
|
-
#
|
287
|
+
# Sets the mode for +self+ to mixed mode
|
288
|
+
# (see {Mixed Mode}[#class-CSV::Table-label-Mixed+Mode]); returns +self+:
|
289
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
290
|
+
# table = CSV.parse(source, headers: true).by_col!
|
291
|
+
# table.mode # => :col
|
292
|
+
# table1 = table.by_col_or_row!
|
293
|
+
# table.mode # => :col_or_row
|
294
|
+
# table1.equal?(table) # => true # Returned self
|
89
295
|
def by_col_or_row!
|
90
296
|
@mode = :col_or_row
|
91
297
|
|
92
298
|
self
|
93
299
|
end
|
94
300
|
|
301
|
+
# :call-seq:
|
302
|
+
# table.by_row -> table_dup
|
95
303
|
#
|
96
|
-
# Returns a duplicate
|
97
|
-
#
|
98
|
-
#
|
304
|
+
# Returns a duplicate of +self+, in row mode
|
305
|
+
# (see {Row Mode}[#class-CSV::Table-label-Row+Mode]):
|
306
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
307
|
+
# table = CSV.parse(source, headers: true)
|
308
|
+
# table.mode # => :col_or_row
|
309
|
+
# dup_table = table.by_row
|
310
|
+
# dup_table.mode # => :row
|
311
|
+
# dup_table.equal?(table) # => false # It's a dup
|
99
312
|
#
|
100
|
-
# This
|
101
|
-
#
|
102
|
-
#
|
313
|
+
# This may be used to chain method calls without changing the mode
|
314
|
+
# (but also will affect performance and memory usage):
|
315
|
+
# dup_table.by_row[1]
|
103
316
|
#
|
317
|
+
# Also note that changes to the duplicate table will not affect the original.
|
104
318
|
def by_row
|
105
319
|
self.class.new(@table.dup).by_row!
|
106
320
|
end
|
107
321
|
|
322
|
+
# :call-seq:
|
323
|
+
# table.by_row! -> self
|
108
324
|
#
|
109
|
-
#
|
110
|
-
#
|
111
|
-
#
|
112
|
-
#
|
113
|
-
#
|
325
|
+
# Sets the mode for +self+ to row mode
|
326
|
+
# (see {Row Mode}[#class-CSV::Table-label-Row+Mode]); returns +self+:
|
327
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
328
|
+
# table = CSV.parse(source, headers: true)
|
329
|
+
# table.mode # => :col_or_row
|
330
|
+
# table1 = table.by_row!
|
331
|
+
# table.mode # => :row
|
332
|
+
# table1.equal?(table) # => true # Returned self
|
114
333
|
def by_row!
|
115
334
|
@mode = :row
|
116
335
|
|
117
336
|
self
|
118
337
|
end
|
119
338
|
|
339
|
+
# :call-seq:
|
340
|
+
# table.headers -> array_of_headers
|
120
341
|
#
|
121
|
-
# Returns
|
122
|
-
# other rows). An empty Array is returned for empty tables.
|
342
|
+
# Returns a new \Array containing the \String headers for the table.
|
123
343
|
#
|
344
|
+
# If the table is not empty, returns the headers from the first row:
|
345
|
+
# rows = [
|
346
|
+
# CSV::Row.new(['Foo', 'Bar'], []),
|
347
|
+
# CSV::Row.new(['FOO', 'BAR'], []),
|
348
|
+
# CSV::Row.new(['foo', 'bar'], []),
|
349
|
+
# ]
|
350
|
+
# table = CSV::Table.new(rows)
|
351
|
+
# table.headers # => ["Foo", "Bar"]
|
352
|
+
# table.delete(0)
|
353
|
+
# table.headers # => ["FOO", "BAR"]
|
354
|
+
# table.delete(0)
|
355
|
+
# table.headers # => ["foo", "bar"]
|
356
|
+
#
|
357
|
+
# If the table is empty, returns a copy of the headers in the table itself:
|
358
|
+
# table.delete(0)
|
359
|
+
# table.headers # => ["Foo", "Bar"]
|
124
360
|
def headers
|
125
361
|
if @table.empty?
|
126
|
-
|
362
|
+
@headers.dup
|
127
363
|
else
|
128
364
|
@table.first.headers
|
129
365
|
end
|
130
366
|
end
|
131
367
|
|
368
|
+
# :call-seq:
|
369
|
+
# table[n] -> row or column_data
|
370
|
+
# table[range] -> array_of_rows or array_of_column_data
|
371
|
+
# table[header] -> array_of_column_data
|
372
|
+
#
|
373
|
+
# Returns data from the table; does not modify the table.
|
374
|
+
#
|
375
|
+
# ---
|
376
|
+
#
|
377
|
+
# Fetch a \Row by Its \Integer Index::
|
378
|
+
# - Form: <tt>table[n]</tt>, +n+ an integer.
|
379
|
+
# - Access mode: <tt>:row</tt> or <tt>:col_or_row</tt>.
|
380
|
+
# - Return value: _nth_ row of the table, if that row exists;
|
381
|
+
# otherwise +nil+.
|
382
|
+
#
|
383
|
+
# Returns the _nth_ row of the table if that row exists:
|
384
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
385
|
+
# table = CSV.parse(source, headers: true)
|
386
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
387
|
+
# table[1] # => #<CSV::Row "Name":"bar" "Value":"1">
|
388
|
+
# table.by_col_or_row! # => #<CSV::Table mode:col_or_row row_count:4>
|
389
|
+
# table[1] # => #<CSV::Row "Name":"bar" "Value":"1">
|
390
|
+
#
|
391
|
+
# Counts backward from the last row if +n+ is negative:
|
392
|
+
# table[-1] # => #<CSV::Row "Name":"baz" "Value":"2">
|
393
|
+
#
|
394
|
+
# Returns +nil+ if +n+ is too large or too small:
|
395
|
+
# table[4] # => nil
|
396
|
+
# table[-4] # => nil
|
397
|
+
#
|
398
|
+
# Raises an exception if the access mode is <tt>:row</tt>
|
399
|
+
# and +n+ is not an \Integer:
|
400
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
401
|
+
# # Raises TypeError (no implicit conversion of String into Integer):
|
402
|
+
# table['Name']
|
403
|
+
#
|
404
|
+
# ---
|
405
|
+
#
|
406
|
+
# Fetch a Column by Its \Integer Index::
|
407
|
+
# - Form: <tt>table[n]</tt>, +n+ an \Integer.
|
408
|
+
# - Access mode: <tt>:col</tt>.
|
409
|
+
# - Return value: _nth_ column of the table, if that column exists;
|
410
|
+
# otherwise an \Array of +nil+ fields of length <tt>self.size</tt>.
|
411
|
+
#
|
412
|
+
# Returns the _nth_ column of the table if that column exists:
|
413
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
414
|
+
# table = CSV.parse(source, headers: true)
|
415
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
416
|
+
# table[1] # => ["0", "1", "2"]
|
417
|
+
#
|
418
|
+
# Counts backward from the last column if +n+ is negative:
|
419
|
+
# table[-2] # => ["foo", "bar", "baz"]
|
420
|
+
#
|
421
|
+
# Returns an \Array of +nil+ fields if +n+ is too large or too small:
|
422
|
+
# table[4] # => [nil, nil, nil]
|
423
|
+
# table[-4] # => [nil, nil, nil]
|
424
|
+
#
|
425
|
+
# ---
|
426
|
+
#
|
427
|
+
# Fetch Rows by \Range::
|
428
|
+
# - Form: <tt>table[range]</tt>, +range+ a \Range object.
|
429
|
+
# - Access mode: <tt>:row</tt> or <tt>:col_or_row</tt>.
|
430
|
+
# - Return value: rows from the table, beginning at row <tt>range.start</tt>,
|
431
|
+
# if those rows exists.
|
432
|
+
#
|
433
|
+
# Returns rows from the table, beginning at row <tt>range.first</tt>,
|
434
|
+
# if those rows exist:
|
435
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
436
|
+
# table = CSV.parse(source, headers: true)
|
437
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
438
|
+
# rows = table[1..2] # => #<CSV::Row "Name":"bar" "Value":"1">
|
439
|
+
# rows # => [#<CSV::Row "Name":"bar" "Value":"1">, #<CSV::Row "Name":"baz" "Value":"2">]
|
440
|
+
# table.by_col_or_row! # => #<CSV::Table mode:col_or_row row_count:4>
|
441
|
+
# rows = table[1..2] # => #<CSV::Row "Name":"bar" "Value":"1">
|
442
|
+
# rows # => [#<CSV::Row "Name":"bar" "Value":"1">, #<CSV::Row "Name":"baz" "Value":"2">]
|
443
|
+
#
|
444
|
+
# If there are too few rows, returns all from <tt>range.start</tt> to the end:
|
445
|
+
# rows = table[1..50] # => #<CSV::Row "Name":"bar" "Value":"1">
|
446
|
+
# rows # => [#<CSV::Row "Name":"bar" "Value":"1">, #<CSV::Row "Name":"baz" "Value":"2">]
|
447
|
+
#
|
448
|
+
# Special case: if <tt>range.start == table.size</tt>, returns an empty \Array:
|
449
|
+
# table[table.size..50] # => []
|
132
450
|
#
|
133
|
-
#
|
134
|
-
#
|
135
|
-
#
|
451
|
+
# If <tt>range.end</tt> is negative, calculates the ending index from the end:
|
452
|
+
# rows = table[0..-1]
|
453
|
+
# rows # => [#<CSV::Row "Name":"foo" "Value":"0">, #<CSV::Row "Name":"bar" "Value":"1">, #<CSV::Row "Name":"baz" "Value":"2">]
|
136
454
|
#
|
137
|
-
#
|
138
|
-
#
|
455
|
+
# If <tt>range.start</tt> is negative, calculates the starting index from the end:
|
456
|
+
# rows = table[-1..2]
|
457
|
+
# rows # => [#<CSV::Row "Name":"baz" "Value":"2">]
|
139
458
|
#
|
459
|
+
# If <tt>range.start</tt> is larger than <tt>table.size</tt>, returns +nil+:
|
460
|
+
# table[4..4] # => nil
|
461
|
+
#
|
462
|
+
# ---
|
463
|
+
#
|
464
|
+
# Fetch Columns by \Range::
|
465
|
+
# - Form: <tt>table[range]</tt>, +range+ a \Range object.
|
466
|
+
# - Access mode: <tt>:col</tt>.
|
467
|
+
# - Return value: column data from the table, beginning at column <tt>range.start</tt>,
|
468
|
+
# if those columns exist.
|
469
|
+
#
|
470
|
+
# Returns column values from the table, if the column exists;
|
471
|
+
# the values are arranged by row:
|
472
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
473
|
+
# table = CSV.parse(source, headers: true)
|
474
|
+
# table.by_col!
|
475
|
+
# table[0..1] # => [["foo", "0"], ["bar", "1"], ["baz", "2"]]
|
476
|
+
#
|
477
|
+
# Special case: if <tt>range.start == headers.size</tt>,
|
478
|
+
# returns an \Array (size: <tt>table.size</tt>) of empty \Arrays:
|
479
|
+
# table[table.headers.size..50] # => [[], [], []]
|
480
|
+
#
|
481
|
+
# If <tt>range.end</tt> is negative, calculates the ending index from the end:
|
482
|
+
# table[0..-1] # => [["foo", "0"], ["bar", "1"], ["baz", "2"]]
|
483
|
+
#
|
484
|
+
# If <tt>range.start</tt> is negative, calculates the starting index from the end:
|
485
|
+
# table[-2..2] # => [["foo", "0"], ["bar", "1"], ["baz", "2"]]
|
486
|
+
#
|
487
|
+
# If <tt>range.start</tt> is larger than <tt>table.size</tt>,
|
488
|
+
# returns an \Array of +nil+ values:
|
489
|
+
# table[4..4] # => [nil, nil, nil]
|
490
|
+
#
|
491
|
+
# ---
|
492
|
+
#
|
493
|
+
# Fetch a Column by Its \String Header::
|
494
|
+
# - Form: <tt>table[header]</tt>, +header+ a \String header.
|
495
|
+
# - Access mode: <tt>:col</tt> or <tt>:col_or_row</tt>
|
496
|
+
# - Return value: column data from the table, if that +header+ exists.
|
497
|
+
#
|
498
|
+
# Returns column values from the table, if the column exists:
|
499
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
500
|
+
# table = CSV.parse(source, headers: true)
|
501
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
502
|
+
# table['Name'] # => ["foo", "bar", "baz"]
|
503
|
+
# table.by_col_or_row! # => #<CSV::Table mode:col_or_row row_count:4>
|
504
|
+
# col = table['Name']
|
505
|
+
# col # => ["foo", "bar", "baz"]
|
506
|
+
#
|
507
|
+
# Modifying the returned column values does not modify the table:
|
508
|
+
# col[0] = 'bat'
|
509
|
+
# col # => ["bat", "bar", "baz"]
|
510
|
+
# table['Name'] # => ["foo", "bar", "baz"]
|
511
|
+
#
|
512
|
+
# Returns an \Array of +nil+ values if there is no such column:
|
513
|
+
# table['Nosuch'] # => [nil, nil, nil]
|
140
514
|
def [](index_or_header)
|
141
515
|
if @mode == :row or # by index
|
142
516
|
(@mode == :col_or_row and (index_or_header.is_a?(Integer) or index_or_header.is_a?(Range)))
|
@@ -146,22 +520,132 @@ class CSV
|
|
146
520
|
end
|
147
521
|
end
|
148
522
|
|
523
|
+
# :call-seq:
|
524
|
+
# table[n] = row -> row
|
525
|
+
# table[n] = field_or_array_of_fields -> field_or_array_of_fields
|
526
|
+
# table[header] = field_or_array_of_fields -> field_or_array_of_fields
|
527
|
+
#
|
528
|
+
# Puts data onto the table.
|
529
|
+
#
|
530
|
+
# ---
|
531
|
+
#
|
532
|
+
# Set a \Row by Its \Integer Index::
|
533
|
+
# - Form: <tt>table[n] = row</tt>, +n+ an \Integer,
|
534
|
+
# +row+ a \CSV::Row instance or an \Array of fields.
|
535
|
+
# - Access mode: <tt>:row</tt> or <tt>:col_or_row</tt>.
|
536
|
+
# - Return value: +row+.
|
149
537
|
#
|
150
|
-
#
|
151
|
-
#
|
152
|
-
#
|
538
|
+
# If the row exists, it is replaced:
|
539
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
540
|
+
# table = CSV.parse(source, headers: true)
|
541
|
+
# new_row = CSV::Row.new(['Name', 'Value'], ['bat', 3])
|
542
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
543
|
+
# return_value = table[0] = new_row
|
544
|
+
# return_value.equal?(new_row) # => true # Returned the row
|
545
|
+
# table[0].to_h # => {"Name"=>"bat", "Value"=>3}
|
153
546
|
#
|
154
|
-
#
|
155
|
-
#
|
547
|
+
# With access mode <tt>:col_or_row</tt>:
|
548
|
+
# table.by_col_or_row! # => #<CSV::Table mode:col_or_row row_count:4>
|
549
|
+
# table[0] = CSV::Row.new(['Name', 'Value'], ['bam', 4])
|
550
|
+
# table[0].to_h # => {"Name"=>"bam", "Value"=>4}
|
156
551
|
#
|
157
|
-
#
|
158
|
-
#
|
159
|
-
#
|
160
|
-
#
|
552
|
+
# With an \Array instead of a \CSV::Row, inherits headers from the table:
|
553
|
+
# array = ['bad', 5]
|
554
|
+
# return_value = table[0] = array
|
555
|
+
# return_value.equal?(array) # => true # Returned the array
|
556
|
+
# table[0].to_h # => {"Name"=>"bad", "Value"=>5}
|
161
557
|
#
|
162
|
-
#
|
163
|
-
#
|
558
|
+
# If the row does not exist, extends the table by adding rows:
|
559
|
+
# assigns rows with +nil+ as needed:
|
560
|
+
# table.size # => 3
|
561
|
+
# table[5] = ['bag', 6]
|
562
|
+
# table.size # => 6
|
563
|
+
# table[3] # => nil
|
564
|
+
# table[4]# => nil
|
565
|
+
# table[5].to_h # => {"Name"=>"bag", "Value"=>6}
|
164
566
|
#
|
567
|
+
# Note that the +nil+ rows are actually +nil+, not a row of +nil+ fields.
|
568
|
+
#
|
569
|
+
# ---
|
570
|
+
#
|
571
|
+
# Set a Column by Its \Integer Index::
|
572
|
+
# - Form: <tt>table[n] = array_of_fields</tt>, +n+ an \Integer,
|
573
|
+
# +array_of_fields+ an \Array of \String fields.
|
574
|
+
# - Access mode: <tt>:col</tt>.
|
575
|
+
# - Return value: +array_of_fields+.
|
576
|
+
#
|
577
|
+
# If the column exists, it is replaced:
|
578
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
579
|
+
# table = CSV.parse(source, headers: true)
|
580
|
+
# new_col = [3, 4, 5]
|
581
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
582
|
+
# return_value = table[1] = new_col
|
583
|
+
# return_value.equal?(new_col) # => true # Returned the column
|
584
|
+
# table[1] # => [3, 4, 5]
|
585
|
+
# # The rows, as revised:
|
586
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
587
|
+
# table[0].to_h # => {"Name"=>"foo", "Value"=>3}
|
588
|
+
# table[1].to_h # => {"Name"=>"bar", "Value"=>4}
|
589
|
+
# table[2].to_h # => {"Name"=>"baz", "Value"=>5}
|
590
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
591
|
+
#
|
592
|
+
# If there are too few values, fills with +nil+ values:
|
593
|
+
# table[1] = [0]
|
594
|
+
# table[1] # => [0, nil, nil]
|
595
|
+
#
|
596
|
+
# If there are too many values, ignores the extra values:
|
597
|
+
# table[1] = [0, 1, 2, 3, 4]
|
598
|
+
# table[1] # => [0, 1, 2]
|
599
|
+
#
|
600
|
+
# If a single value is given, replaces all fields in the column with that value:
|
601
|
+
# table[1] = 'bat'
|
602
|
+
# table[1] # => ["bat", "bat", "bat"]
|
603
|
+
#
|
604
|
+
# ---
|
605
|
+
#
|
606
|
+
# Set a Column by Its \String Header::
|
607
|
+
# - Form: <tt>table[header] = field_or_array_of_fields</tt>,
|
608
|
+
# +header+ a \String header, +field_or_array_of_fields+ a field value
|
609
|
+
# or an \Array of \String fields.
|
610
|
+
# - Access mode: <tt>:col</tt> or <tt>:col_or_row</tt>.
|
611
|
+
# - Return value: +field_or_array_of_fields+.
|
612
|
+
#
|
613
|
+
# If the column exists, it is replaced:
|
614
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
615
|
+
# table = CSV.parse(source, headers: true)
|
616
|
+
# new_col = [3, 4, 5]
|
617
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
618
|
+
# return_value = table['Value'] = new_col
|
619
|
+
# return_value.equal?(new_col) # => true # Returned the column
|
620
|
+
# table['Value'] # => [3, 4, 5]
|
621
|
+
# # The rows, as revised:
|
622
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
623
|
+
# table[0].to_h # => {"Name"=>"foo", "Value"=>3}
|
624
|
+
# table[1].to_h # => {"Name"=>"bar", "Value"=>4}
|
625
|
+
# table[2].to_h # => {"Name"=>"baz", "Value"=>5}
|
626
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
627
|
+
#
|
628
|
+
# If there are too few values, fills with +nil+ values:
|
629
|
+
# table['Value'] = [0]
|
630
|
+
# table['Value'] # => [0, nil, nil]
|
631
|
+
#
|
632
|
+
# If there are too many values, ignores the extra values:
|
633
|
+
# table['Value'] = [0, 1, 2, 3, 4]
|
634
|
+
# table['Value'] # => [0, 1, 2]
|
635
|
+
#
|
636
|
+
# If the column does not exist, extends the table by adding columns:
|
637
|
+
# table['Note'] = ['x', 'y', 'z']
|
638
|
+
# table['Note'] # => ["x", "y", "z"]
|
639
|
+
# # The rows, as revised:
|
640
|
+
# table.by_row!
|
641
|
+
# table[0].to_h # => {"Name"=>"foo", "Value"=>0, "Note"=>"x"}
|
642
|
+
# table[1].to_h # => {"Name"=>"bar", "Value"=>1, "Note"=>"y"}
|
643
|
+
# table[2].to_h # => {"Name"=>"baz", "Value"=>2, "Note"=>"z"}
|
644
|
+
# table.by_col!
|
645
|
+
#
|
646
|
+
# If a single value is given, replaces all fields in the column with that value:
|
647
|
+
# table['Value'] = 'bat'
|
648
|
+
# table['Value'] # => ["bat", "bat", "bat"]
|
165
649
|
def []=(index_or_header, value)
|
166
650
|
if @mode == :row or # by index
|
167
651
|
(@mode == :col_or_row and index_or_header.is_a? Integer)
|
@@ -171,6 +655,10 @@ class CSV
|
|
171
655
|
@table[index_or_header] = value
|
172
656
|
end
|
173
657
|
else # set column
|
658
|
+
unless index_or_header.is_a? Integer
|
659
|
+
index = @headers.index(index_or_header) || @headers.size
|
660
|
+
@headers[index] = index_or_header
|
661
|
+
end
|
174
662
|
if value.is_a? Array # multiple values
|
175
663
|
@table.each_with_index do |row, i|
|
176
664
|
if row.header_row?
|
@@ -191,15 +679,58 @@ class CSV
|
|
191
679
|
end
|
192
680
|
end
|
193
681
|
|
682
|
+
# :call-seq:
|
683
|
+
# table.values_at(*indexes) -> array_of_rows
|
684
|
+
# table.values_at(*headers) -> array_of_columns_data
|
685
|
+
#
|
686
|
+
# If the access mode is <tt>:row</tt> or <tt>:col_or_row</tt>,
|
687
|
+
# and each argument is either an \Integer or a \Range,
|
688
|
+
# returns rows.
|
689
|
+
# Otherwise, returns columns data.
|
194
690
|
#
|
195
|
-
#
|
196
|
-
#
|
197
|
-
# access. For columnar access, the return set has an Array for each row
|
198
|
-
# with the values indicated by the headers in each Array. You can force
|
199
|
-
# column or row mode using by_col!() or by_row!().
|
691
|
+
# In either case, the returned values are in the order
|
692
|
+
# specified by the arguments. Arguments may be repeated.
|
200
693
|
#
|
201
|
-
#
|
694
|
+
# ---
|
202
695
|
#
|
696
|
+
# Returns rows as an \Array of \CSV::Row objects.
|
697
|
+
#
|
698
|
+
# No argument:
|
699
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
700
|
+
# table = CSV.parse(source, headers: true)
|
701
|
+
# table.values_at # => []
|
702
|
+
#
|
703
|
+
# One index:
|
704
|
+
# values = table.values_at(0)
|
705
|
+
# values # => [#<CSV::Row "Name":"foo" "Value":"0">]
|
706
|
+
#
|
707
|
+
# Two indexes:
|
708
|
+
# values = table.values_at(2, 0)
|
709
|
+
# values # => [#<CSV::Row "Name":"baz" "Value":"2">, #<CSV::Row "Name":"foo" "Value":"0">]
|
710
|
+
#
|
711
|
+
# One \Range:
|
712
|
+
# values = table.values_at(1..2)
|
713
|
+
# values # => [#<CSV::Row "Name":"bar" "Value":"1">, #<CSV::Row "Name":"baz" "Value":"2">]
|
714
|
+
#
|
715
|
+
# \Ranges and indexes:
|
716
|
+
# values = table.values_at(0..1, 1..2, 0, 2)
|
717
|
+
# pp values
|
718
|
+
# Output:
|
719
|
+
# [#<CSV::Row "Name":"foo" "Value":"0">,
|
720
|
+
# #<CSV::Row "Name":"bar" "Value":"1">,
|
721
|
+
# #<CSV::Row "Name":"bar" "Value":"1">,
|
722
|
+
# #<CSV::Row "Name":"baz" "Value":"2">,
|
723
|
+
# #<CSV::Row "Name":"foo" "Value":"0">,
|
724
|
+
# #<CSV::Row "Name":"baz" "Value":"2">]
|
725
|
+
#
|
726
|
+
# ---
|
727
|
+
#
|
728
|
+
# Returns columns data as row Arrays,
|
729
|
+
# each consisting of the specified columns data for that row:
|
730
|
+
# values = table.values_at('Name')
|
731
|
+
# values # => [["foo"], ["bar"], ["baz"]]
|
732
|
+
# values = table.values_at('Value', 'Name')
|
733
|
+
# values # => [["0", "foo"], ["1", "bar"], ["2", "baz"]]
|
203
734
|
def values_at(*indices_or_headers)
|
204
735
|
if @mode == :row or # by indices
|
205
736
|
( @mode == :col_or_row and indices_or_headers.all? do |index|
|
@@ -214,13 +745,20 @@ class CSV
|
|
214
745
|
end
|
215
746
|
end
|
216
747
|
|
748
|
+
# :call-seq:
|
749
|
+
# table << row_or_array -> self
|
217
750
|
#
|
218
|
-
#
|
219
|
-
#
|
220
|
-
#
|
221
|
-
#
|
222
|
-
#
|
751
|
+
# If +row_or_array+ is a \CSV::Row object,
|
752
|
+
# it is appended to the table:
|
753
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
754
|
+
# table = CSV.parse(source, headers: true)
|
755
|
+
# table << CSV::Row.new(table.headers, ['bat', 3])
|
756
|
+
# table[3] # => #<CSV::Row "Name":"bat" "Value":3>
|
223
757
|
#
|
758
|
+
# If +row_or_array+ is an \Array, it is used to create a new
|
759
|
+
# \CSV::Row object which is then appended to the table:
|
760
|
+
# table << ['bam', 4]
|
761
|
+
# table[4] # => #<CSV::Row "Name":"bam" "Value":4>
|
224
762
|
def <<(row_or_array)
|
225
763
|
if row_or_array.is_a? Array # append Array
|
226
764
|
@table << Row.new(headers, row_or_array)
|
@@ -232,23 +770,67 @@ class CSV
|
|
232
770
|
end
|
233
771
|
|
234
772
|
#
|
235
|
-
#
|
773
|
+
# :call-seq:
|
774
|
+
# table.push(*rows_or_arrays) -> self
|
236
775
|
#
|
237
|
-
#
|
238
|
-
#
|
239
|
-
# This method returns the table for chaining.
|
776
|
+
# A shortcut for appending multiple rows. Equivalent to:
|
777
|
+
# rows.each {|row| self << row }
|
240
778
|
#
|
779
|
+
# Each argument may be either a \CSV::Row object or an \Array:
|
780
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
781
|
+
# table = CSV.parse(source, headers: true)
|
782
|
+
# rows = [
|
783
|
+
# CSV::Row.new(table.headers, ['bat', 3]),
|
784
|
+
# ['bam', 4]
|
785
|
+
# ]
|
786
|
+
# table.push(*rows)
|
787
|
+
# table[3..4] # => [#<CSV::Row "Name":"bat" "Value":3>, #<CSV::Row "Name":"bam" "Value":4>]
|
241
788
|
def push(*rows)
|
242
789
|
rows.each { |row| self << row }
|
243
790
|
|
244
791
|
self # for chaining
|
245
792
|
end
|
246
793
|
|
794
|
+
# :call-seq:
|
795
|
+
# table.delete(*indexes) -> deleted_values
|
796
|
+
# table.delete(*headers) -> deleted_values
|
797
|
+
#
|
798
|
+
# If the access mode is <tt>:row</tt> or <tt>:col_or_row</tt>,
|
799
|
+
# and each argument is either an \Integer or a \Range,
|
800
|
+
# returns deleted rows.
|
801
|
+
# Otherwise, returns deleted columns data.
|
802
|
+
#
|
803
|
+
# In either case, the returned values are in the order
|
804
|
+
# specified by the arguments. Arguments may be repeated.
|
805
|
+
#
|
806
|
+
# ---
|
807
|
+
#
|
808
|
+
# Returns rows as an \Array of \CSV::Row objects.
|
809
|
+
#
|
810
|
+
# One index:
|
811
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
812
|
+
# table = CSV.parse(source, headers: true)
|
813
|
+
# deleted_values = table.delete(0)
|
814
|
+
# deleted_values # => [#<CSV::Row "Name":"foo" "Value":"0">]
|
815
|
+
#
|
816
|
+
# Two indexes:
|
817
|
+
# table = CSV.parse(source, headers: true)
|
818
|
+
# deleted_values = table.delete(2, 0)
|
819
|
+
# deleted_values # => [#<CSV::Row "Name":"baz" "Value":"2">, #<CSV::Row "Name":"foo" "Value":"0">]
|
247
820
|
#
|
248
|
-
#
|
249
|
-
# mode indices refer to rows and everything else is assumed to be a column
|
250
|
-
# headers. Use by_col!() or by_row!() to force the lookup.
|
821
|
+
# ---
|
251
822
|
#
|
823
|
+
# Returns columns data as column Arrays.
|
824
|
+
#
|
825
|
+
# One header:
|
826
|
+
# table = CSV.parse(source, headers: true)
|
827
|
+
# deleted_values = table.delete('Name')
|
828
|
+
# deleted_values # => ["foo", "bar", "baz"]
|
829
|
+
#
|
830
|
+
# Two headers:
|
831
|
+
# table = CSV.parse(source, headers: true)
|
832
|
+
# deleted_values = table.delete('Value', 'Name')
|
833
|
+
# deleted_values # => [["0", "1", "2"], ["foo", "bar", "baz"]]
|
252
834
|
def delete(*indexes_or_headers)
|
253
835
|
if indexes_or_headers.empty?
|
254
836
|
raise ArgumentError, "wrong number of arguments (given 0, expected 1+)"
|
@@ -258,6 +840,11 @@ class CSV
|
|
258
840
|
(@mode == :col_or_row and index_or_header.is_a? Integer)
|
259
841
|
@table.delete_at(index_or_header)
|
260
842
|
else # by header
|
843
|
+
if index_or_header.is_a? Integer
|
844
|
+
@headers.delete_at(index_or_header)
|
845
|
+
else
|
846
|
+
@headers.delete(index_or_header)
|
847
|
+
end
|
261
848
|
@table.map { |row| row.delete(index_or_header).last }
|
262
849
|
end
|
263
850
|
end
|
@@ -268,25 +855,43 @@ class CSV
|
|
268
855
|
end
|
269
856
|
end
|
270
857
|
|
858
|
+
# :call-seq:
|
859
|
+
# table.delete_if {|row_or_column| ... } -> self
|
271
860
|
#
|
272
|
-
# Removes
|
273
|
-
#
|
274
|
-
# walking of rows. In column mode, iteration will +yield+ two element
|
275
|
-
# tuples containing the column name and an Array of values for that column.
|
861
|
+
# Removes rows or columns for which the block returns a truthy value;
|
862
|
+
# returns +self+.
|
276
863
|
#
|
277
|
-
#
|
864
|
+
# Removes rows when the access mode is <tt>:row</tt> or <tt>:col_or_row</tt>;
|
865
|
+
# calls the block with each \CSV::Row object:
|
866
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
867
|
+
# table = CSV.parse(source, headers: true)
|
868
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
869
|
+
# table.size # => 3
|
870
|
+
# table.delete_if {|row| row['Name'].start_with?('b') }
|
871
|
+
# table.size # => 1
|
278
872
|
#
|
279
|
-
#
|
873
|
+
# Removes columns when the access mode is <tt>:col</tt>;
|
874
|
+
# calls the block with each column as a 2-element array
|
875
|
+
# containing the header and an \Array of column fields:
|
876
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
877
|
+
# table = CSV.parse(source, headers: true)
|
878
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
879
|
+
# table.headers.size # => 2
|
880
|
+
# table.delete_if {|column_data| column_data[1].include?('2') }
|
881
|
+
# table.headers.size # => 1
|
280
882
|
#
|
883
|
+
# Returns a new \Enumerator if no block is given:
|
884
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
885
|
+
# table = CSV.parse(source, headers: true)
|
886
|
+
# table.delete_if # => #<Enumerator: #<CSV::Table mode:col_or_row row_count:4>:delete_if>
|
281
887
|
def delete_if(&block)
|
282
888
|
return enum_for(__method__) { @mode == :row or @mode == :col_or_row ? size : headers.size } unless block_given?
|
283
889
|
|
284
890
|
if @mode == :row or @mode == :col_or_row # by index
|
285
891
|
@table.delete_if(&block)
|
286
892
|
else # by header
|
287
|
-
deleted = []
|
288
893
|
headers.each do |header|
|
289
|
-
|
894
|
+
delete(header) if yield([header, self[header]])
|
290
895
|
end
|
291
896
|
end
|
292
897
|
|
@@ -295,20 +900,40 @@ class CSV
|
|
295
900
|
|
296
901
|
include Enumerable
|
297
902
|
|
903
|
+
# :call-seq:
|
904
|
+
# table.each {|row_or_column| ... ) -> self
|
298
905
|
#
|
299
|
-
#
|
300
|
-
# walking of rows. In column mode, iteration will +yield+ two element
|
301
|
-
# tuples containing the column name and an Array of values for that column.
|
906
|
+
# Calls the block with each row or column; returns +self+.
|
302
907
|
#
|
303
|
-
#
|
908
|
+
# When the access mode is <tt>:row</tt> or <tt>:col_or_row</tt>,
|
909
|
+
# calls the block with each \CSV::Row object:
|
910
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
911
|
+
# table = CSV.parse(source, headers: true)
|
912
|
+
# table.by_row! # => #<CSV::Table mode:row row_count:4>
|
913
|
+
# table.each {|row| p row }
|
914
|
+
# Output:
|
915
|
+
# #<CSV::Row "Name":"foo" "Value":"0">
|
916
|
+
# #<CSV::Row "Name":"bar" "Value":"1">
|
917
|
+
# #<CSV::Row "Name":"baz" "Value":"2">
|
304
918
|
#
|
305
|
-
#
|
919
|
+
# When the access mode is <tt>:col</tt>,
|
920
|
+
# calls the block with each column as a 2-element array
|
921
|
+
# containing the header and an \Array of column fields:
|
922
|
+
# table.by_col! # => #<CSV::Table mode:col row_count:4>
|
923
|
+
# table.each {|column_data| p column_data }
|
924
|
+
# Output:
|
925
|
+
# ["Name", ["foo", "bar", "baz"]]
|
926
|
+
# ["Value", ["0", "1", "2"]]
|
306
927
|
#
|
928
|
+
# Returns a new \Enumerator if no block is given:
|
929
|
+
# table.each # => #<Enumerator: #<CSV::Table mode:col row_count:4>:each>
|
307
930
|
def each(&block)
|
308
931
|
return enum_for(__method__) { @mode == :col ? headers.size : size } unless block_given?
|
309
932
|
|
310
933
|
if @mode == :col
|
311
|
-
headers.each
|
934
|
+
headers.each.with_index do |header, i|
|
935
|
+
yield([header, @table.map {|row| row[header, i]}])
|
936
|
+
end
|
312
937
|
else
|
313
938
|
@table.each(&block)
|
314
939
|
end
|
@@ -316,16 +941,40 @@ class CSV
|
|
316
941
|
self # for chaining
|
317
942
|
end
|
318
943
|
|
319
|
-
#
|
944
|
+
# :call-seq:
|
945
|
+
# table == other_table -> true or false
|
946
|
+
#
|
947
|
+
# Returns +true+ if all each row of +self+ <tt>==</tt>
|
948
|
+
# the corresponding row of +other_table+, otherwise, +false+.
|
949
|
+
#
|
950
|
+
# The access mode does no affect the result.
|
951
|
+
#
|
952
|
+
# Equal tables:
|
953
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
954
|
+
# table = CSV.parse(source, headers: true)
|
955
|
+
# other_table = CSV.parse(source, headers: true)
|
956
|
+
# table == other_table # => true
|
957
|
+
#
|
958
|
+
# Different row count:
|
959
|
+
# other_table.delete(2)
|
960
|
+
# table == other_table # => false
|
961
|
+
#
|
962
|
+
# Different last row:
|
963
|
+
# other_table << ['bat', 3]
|
964
|
+
# table == other_table # => false
|
320
965
|
def ==(other)
|
321
966
|
return @table == other.table if other.is_a? CSV::Table
|
322
967
|
@table == other
|
323
968
|
end
|
324
969
|
|
970
|
+
# :call-seq:
|
971
|
+
# table.to_a -> array_of_arrays
|
325
972
|
#
|
326
|
-
# Returns the table as an Array of Arrays
|
327
|
-
#
|
328
|
-
#
|
973
|
+
# Returns the table as an \Array of \Arrays;
|
974
|
+
# the headers are in the first row:
|
975
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
976
|
+
# table = CSV.parse(source, headers: true)
|
977
|
+
# table.to_a # => [["Name", "Value"], ["foo", "0"], ["bar", "1"], ["baz", "2"]]
|
329
978
|
def to_a
|
330
979
|
array = [headers]
|
331
980
|
@table.each do |row|
|
@@ -335,17 +984,30 @@ class CSV
|
|
335
984
|
array
|
336
985
|
end
|
337
986
|
|
987
|
+
# :call-seq:
|
988
|
+
# table.to_csv(**options) -> csv_string
|
338
989
|
#
|
339
|
-
# Returns the table as
|
340
|
-
#
|
990
|
+
# Returns the table as \CSV string.
|
991
|
+
# See {Options for Generating}[../CSV.html#class-CSV-label-Options+for+Generating].
|
341
992
|
#
|
342
|
-
#
|
343
|
-
#
|
993
|
+
# Defaults option +write_headers+ to +true+:
|
994
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
995
|
+
# table = CSV.parse(source, headers: true)
|
996
|
+
# table.to_csv # => "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
344
997
|
#
|
345
|
-
|
346
|
-
|
347
|
-
|
348
|
-
|
998
|
+
# Omits the headers if option +write_headers+ is given as +false+
|
999
|
+
# (see {Option +write_headers+}[../CSV.html#class-CSV-label-Option+write_headers]):
|
1000
|
+
# table.to_csv(write_headers: false) # => "foo,0\nbar,1\nbaz,2\n"
|
1001
|
+
#
|
1002
|
+
# Limit rows if option +limit+ is given like +2+:
|
1003
|
+
# table.to_csv(limit: 2) # => "Name,Value\nfoo,0\nbar,1\n"
|
1004
|
+
def to_csv(write_headers: true, limit: nil, **options)
|
1005
|
+
array = write_headers ? [headers.to_csv(**options)] : []
|
1006
|
+
limit ||= @table.size
|
1007
|
+
limit = @table.size + 1 + limit if limit < 0
|
1008
|
+
limit = 0 if limit < 0
|
1009
|
+
@table.first(limit).each do |row|
|
1010
|
+
array.push(row.fields.to_csv(**options)) unless row.header_row?
|
349
1011
|
end
|
350
1012
|
|
351
1013
|
array.join("")
|
@@ -370,9 +1032,24 @@ class CSV
|
|
370
1032
|
end
|
371
1033
|
end
|
372
1034
|
|
373
|
-
#
|
1035
|
+
# :call-seq:
|
1036
|
+
# table.inspect => string
|
1037
|
+
#
|
1038
|
+
# Returns a <tt>US-ASCII</tt>-encoded \String showing table:
|
1039
|
+
# - Class: <tt>CSV::Table</tt>.
|
1040
|
+
# - Access mode: <tt>:row</tt>, <tt>:col</tt>, or <tt>:col_or_row</tt>.
|
1041
|
+
# - Size: Row count, including the header row.
|
1042
|
+
#
|
1043
|
+
# Example:
|
1044
|
+
# source = "Name,Value\nfoo,0\nbar,1\nbaz,2\n"
|
1045
|
+
# table = CSV.parse(source, headers: true)
|
1046
|
+
# table.inspect # => "#<CSV::Table mode:col_or_row row_count:4>\nName,Value\nfoo,0\nbar,1\nbaz,2\n"
|
1047
|
+
#
|
374
1048
|
def inspect
|
375
|
-
"#<#{self.class} mode:#{@mode} row_count:#{to_a.size}>"
|
1049
|
+
inspected = +"#<#{self.class} mode:#{@mode} row_count:#{to_a.size}>"
|
1050
|
+
summary = to_csv(limit: 5)
|
1051
|
+
inspected << "\n" << summary if summary.encoding.ascii_compatible?
|
1052
|
+
inspected
|
376
1053
|
end
|
377
1054
|
end
|
378
|
-
end
|
1055
|
+
end
|