Skip to content

Instantly share code, notes, and snippets.

@zHaytam
Created December 19, 2019 14:29
Show Gist options
  • Save zHaytam/f2a30eeb91b92981e9020dda91ea2697 to your computer and use it in GitHub Desktop.
Save zHaytam/f2a30eeb91b92981e9020dda91ea2697 to your computer and use it in GitHub Desktop.
Some utility functions for the Microsoft.Data.Analysis.DataFrame class
public static class DataFrameUtils
{
public static void PrettyPrint(this DataFrame df)
{
var sb = new StringBuilder();
int width = GetLongestValueLength(df) + 4;
for (int i = 0; i < df.Columns.Count; i++)
{
// Left align by 10
sb.Append(string.Format(df.Columns[i].Name.PadRight(width)));
}
sb.AppendLine();
long numberOfRows = Math.Min(df.Rows.Count, 25);
for (int i = 0; i < numberOfRows; i++)
{
foreach (object obj in df.Rows[i])
{
sb.Append((obj ?? "null").ToString().PadRight(width));
}
sb.AppendLine();
}
Console.WriteLine(sb.ToString());
}
private static int GetLongestValueLength(DataFrame df)
{
long numberOfRows = Math.Min(df.Rows.Count, 25);
int longestValueLength = 0;
for (int i = 0; i < numberOfRows; i++)
{
foreach (var value in df.Rows[i])
longestValueLength = Math.Max(longestValueLength, value?.ToString().Length ?? 0);
}
return longestValueLength;
}
public static PrimitiveDataFrameColumn<TResult> Apply<T, TResult>(this PrimitiveDataFrameColumn<T> column,
Func<T, TResult> func)
where T : unmanaged
where TResult : unmanaged
{
var resultColumn = new PrimitiveDataFrameColumn<TResult>(string.Empty, 0);
foreach (var row in column)
resultColumn.Append(func(row.Value));
return resultColumn;
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment