我在 .csv 文件中有一个树结构（节点是文本类型），在读取 csv 后，我想将数据存储在 ruby 对象中。我浏览了一些树插件，我认为nested_set 会为我服务。 但是，我在修复 csv 文件格式时遇到问题，以便我可以读取它并转换为树对象。有没有直接将csv文件或二维数组转换成树形数据结构的方法？？
【问题讨论】：
标签： ruby csv ruby-on-rails-plugins multidimensional-array nested-sets
在您澄清不需要将此树存储在数据库中之后，我建议丢弃 NestedSets（它们用于在 RDBMS 上存储嵌套的对象集，您不需要）。你需要什么我简单的树
class Node
attr_accessor :parent, :children, :text
def initialize(text)
@text = text
@children = []
end
end
由于我有权选择 CSV 文件的格式，所以我建议这样：
id,parent,text
1,,"1"
2,1,"1.1"
3,1,"1.2"
3,2,"1.1.1"
树根是第一行，没有父级，并且总是有父级在其子级之前声明的顺序。这样你就可以建树了
def build_tree(rows)
nodes = {}
rows.each do |row|
node = Node.new(row[:text])
nodes[row[:id]] = node
node.parent = nodes[row[:parent]]
nodes[row[:parent]].children << node if row[:parent]
end
nodes.values.find {|node| node.parent.nil? }
end
root = build_tree(rows)
root.text #=> "1"
root.children.map(&:text) #=> ["1.1", "1.2"]
root.children[0].children.map(&:text) #=> ["1.1.1"]
如果你需要从子节点中获取所有文本，那么你需要使用更多的技巧
def get_nodes(tree_node)
[ tree_node, tree_node.children.map{|node| get_nodes(node)} ].flatten
end
get_nodes(root).map(&:text) #=> ["1", "1.1", "1.1.1", "1.2"]
【讨论】：
看来您根本不需要使用 ORM。你为什么不自己做你的树逻辑，用像Ruby这样的动态语言，这很容易：
require 'set'
# expects an array of [parent, child] pairs, returns the root element of a tree
def make_tree a
tree = {}
a.each do |p, c|
tree[p] ||= {:value => p}
tree[p][:children] ||= Set.new
tree[c] ||= {:value => c}
tree[c][:parent] = tree[p]
tree[p][:children] << tree[c]
end
tree.values.find{|e| e[:parent].nil?}
end
root = make_tree [[1,2],[3,4],[1,3],[4,5]]
puts root.inspect
puts root[:value]
或者，如果你想要更多的面向对象，你可以使用TreeNode 类而不是上面的Hash。
哦，如果您需要直接通过键（在本例中为整数值本身）访问特定树节点，请将方法更改为返回 tree 哈希而不是仅返回根元素。
【讨论】：