JSON到CSV，跳过某些列并重新排序其他列 - Ruby
我有一个很好地将JSON文件转换为CSV文件的工作脚本，但是我正在尝试编辑脚本以在保存之前对CSV文件进行一些更改，目前没有任何成功。
这是我目前的转换脚本：
require 'csv'
require 'json'
require 'set'
def get_recursive_keys(hash, nested_key=nil)
hash.each_with_object([]) do |(k,v),keys|
k = "#{nested_key}.#{k}" unless nested_key.nil?
if v.is_a? Hash
keys.concat(get_recursive_keys(v, k))
else
keys << k
endend
end
json = JSON.parse(File.open(ARGV[0]).read)
headings = Set.new
json.each do |hash|
headings.merge(get_recursive_keys(hash))
end
headings = headings.to_a
CSV.open(ARGV[0] + '.csv', 'w') do |csv|
csv << headings
json.each do |hash|
row = headings.map do |h|
v = hash.dig(*h.split('.'))
v.is_a?(Array) ? v.join(',') : v
end
csv << rowend
end
我用这个命令运行：
for file in directory/*; do ruby json-to-csv.rb "$file"; done
如何编辑此脚本以：
删除包含特定标题的列，例如“score”和“original_name”
（将剩余的列从左到右按字母顺序重新排序） - 如果可能的话？
到目前为止，我所尝试的一切都完全破坏了脚本 - 哪里是开始进行这些更改的最佳位置？
写回答
-
这是有效的代码：
require 'csv'
require 'json'
require 'set'def get_recursive_keys(hash, nested_key=nil)
hash.each_with_object([]) do |(k,v),keys|# Col filter next if ["score", "original_name"].include? k k = "#{nested_key}.#{k}" unless nested_key.nil? if v.is_a? Hash keys.concat(get_recursive_keys(v, k)) else keys << k endend
endjson = JSON.parse(File.open(ARGV[0]).read)
headings = Set.new
headings = get_recursive_keys(json)headings = headings.to_a
Header sorting
headings = headings.sort { |a, b| a <=> b }
CSV.open(ARGV[0] + '.csv', 'w') do |csv|
csv << headings
row = headings.map do |h|v = (h.split('.').length > 1) ? json.dig(*h.split('.')) : h v.is_a?(Array) ? v.join(',') : vend
csv << row
end
我测试了这个小的json字符串：{“score”：“12”，“name”：“Obi”，“original_name”：“Wan Kenobi”，“something”：{“sub_key”：“Wuhu”}}2019-07-17 23:33:06赞同 展开评论 打赏
版权声明：本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
