added better accessor for []=

Author: v0dro

lib/daru/dataframe.rb

@@ -187,13 +187,18 @@ def [](*names)
     end
 
     # Insert a new row/vector of the specified name or modify a previous row.
-    #   Instead of using this method directly, use df.row[:a] = [1,2,3] to set/create
-    #   a row ':a' to [1,2,3], or df.vector[:vec] = [1,2,3] for vectors.
+    # Instead of using this method directly, use df.row[:a] = [1,2,3] to set/create
+    # a row ':a' to [1,2,3], or df.vector[:vec] = [1,2,3] for vectors.
     # 
     # In case a Daru::Vector is specified after the equality the sign, the indexes
-    #   of the vector will be matched against the row/vector indexes of the DataFrame
-    #   before an insertion is performed. Unmatched indexes will be set to nil.
-    def []=(name, axis ,vector)
+    # of the vector will be matched against the row/vector indexes of the DataFrame
+    # before an insertion is performed. Unmatched indexes will be set to nil.
+    def []=(*args)
+      name   = args[0]
+      axis   = args[1]
+      vector = args[-1]
+
+      axis = (!axis.is_a?(Symbol) and (axis != :vector or axis != :row)) ? :vector : axis
       if axis == :vector
         insert_or_modify_vector name, vector
       elsif axis == :row        

lib/daru/vector.rb

@@ -383,8 +383,8 @@ def not_nil?
     # 
     # * +replacement+ - The value which should replace all nils
     def replace_nils! replacement
-      @index.each do |idx|
-        self[idx] = replacement if self[idx].nil?
+      nil_positions.each do |idx|
+        self[idx] = replacement
       end
 
       self

spec/dataframe_spec.rb

@@ -488,6 +488,23 @@
     end
   end
 
+  context "#[]=" do
+    context Daru::Index do
+      it "assigns directly with the []= operator" do
+        @data_frame[:a] = [100,200,300,400,500]
+        expect(@data_frame).to eq(Daru::DataFrame.new({
+          b: [11,12,13,14,15], 
+          a: [100,200,300,400,500], 
+          c: [11,22,33,44,55]}, order: [:a, :b, :c], 
+          index: [:one, :two, :three, :four, :five]))
+      end
+    end
+
+    context Daru::MultiIndex do
+      pending
+    end
+  end
+
   context "#[:row]=" do
     context Daru::Index do
       before :each do