Add VBufferDataFrameCoumn to DataFrame

src/Microsoft.Data.Analysis/IDataView.Extension.cs

@@ -116,6 +116,10 @@ public static DataFrame ToDataFrame(this IDataView dataView, long maxRows, param
                 {
                     dataFrameColumns.Add(new StringDataFrameColumn(dataViewColumn.Name));
                 }
+                else if (type is VectorDataViewType vectorType)
+                {
+                    dataFrameColumns.Add(GetVectorDataFrame(vectorType, dataViewColumn.Name));
+                }
                 else
                 {
                     throw new NotSupportedException(String.Format(Microsoft.Data.Strings.NotSupportedColumnType, type.RawType.Name));
@@ -143,6 +147,66 @@ public static DataFrame ToDataFrame(this IDataView dataView, long maxRows, param
 
             return new DataFrame(dataFrameColumns);
         }
+
+        private static DataFrameColumn GetVectorDataFrame(VectorDataViewType vectorType, string name)
+        {
+            var itemType = vectorType.ItemType;
+
+            if (itemType.RawType == typeof(bool))
+            {
+                return new VBufferDataFrameColumn<bool>(name);
+            }
+            else if (itemType.RawType == typeof(byte))
+            {
+                return new VBufferDataFrameColumn<byte>(name);
+            }
+            else if (itemType.RawType == typeof(double))
+            {
+                return new VBufferDataFrameColumn<double>(name);
+            }
+            else if (itemType.RawType == typeof(float))
+            {
+                return new VBufferDataFrameColumn<Single>(name);
+            }
+            else if (itemType.RawType == typeof(int))
+            {
+                return new VBufferDataFrameColumn<Int32>(name);
+            }
+            else if (itemType.RawType == typeof(long))
+            {
+                return new VBufferDataFrameColumn<Int64>(name);
+            }
+            else if (itemType.RawType == typeof(sbyte))
+            {
+                return new VBufferDataFrameColumn<sbyte>(name);
+            }
+            else if (itemType.RawType == typeof(short))
+            {
+                return new VBufferDataFrameColumn<short>(name);
+            }
+            else if (itemType.RawType == typeof(uint))
+            {
+                return new VBufferDataFrameColumn<uint>(name);
+            }
+            else if (itemType.RawType == typeof(ulong))
+            {
+                return new VBufferDataFrameColumn<ulong>(name);
+            }
+            else if (itemType.RawType == typeof(ushort))
+            {
+                return new VBufferDataFrameColumn<ushort>(name);
+            }
+            else if (itemType.RawType == typeof(char))
+            {
+                return new VBufferDataFrameColumn<char>(name);
+            }
+            else if (itemType.RawType == typeof(decimal))
+            {
+                return new VBufferDataFrameColumn<decimal>(name);
+            }
+
+            throw new NotSupportedException("Specified vector subtype " + itemType.ToString() + " is not supported.");
+        }
     }
 
 }