/// <summary>
/// Get the dataframe for the columns specified
/// </summary>
/// <param name="columnNumbers">The column numbers.</param>
/// <returns></returns>
public DataFrame GetFrame(params int[] columnNumbers)
{
DataFrame result = new DataFrame();
List <float> xFrameRow = new List <float>();
Data.ForEach((record) =>
{
xFrameRow = new List <float>();
for (int i = 0; i < Columns.Count; i++)
{
if (columnNumbers.Contains(i))
{
xFrameRow.Add(record[i]);
}
}
result.Add(xFrameRow);
});
return(result);
}
/// <summary>
/// Get the dataframe for the columns specified
/// </summary>
/// <param name="columnNames">The column names.</param>
/// <returns></returns>
public DataFrame GetFrame(params string[] columnNames)
{
DataFrame result = new DataFrame();
List <float> xFrameRow = new List <float>();
Data.ForEach((record) =>
{
xFrameRow = new List <float>();
foreach (var col in Columns)
{
if (columnNames.Contains(col))
{
xFrameRow.Add(record[Columns.IndexOf(col)]);
result.Columns.Add(col);
}
}
result.Add(xFrameRow);
});
return(result);
}