in spark/sql/dataframe.go [827:849]
func (df *dataFrameImpl) Drop(ctx context.Context, columns ...column.Convertible) (DataFrame, error) {
exprs := make([]*proto.Expression, 0, len(columns))
for _, c := range columns {
e, err := c.ToProto(ctx)
if err != nil {
return nil, err
}
exprs = append(exprs, e)
}
rel := &proto.Relation{
Common: &proto.RelationCommon{
PlanId: newPlanId(),
},
RelType: &proto.Relation_Drop{
Drop: &proto.Drop{
Input: df.relation,
Columns: exprs,
},
},
}
return NewDataFrame(df.session, rel), nil
}