magnus167/rustframe
1
0
Fork 0
mirror of https://github.com/Magnus167/rustframe.git synced 2025-08-20 09:30:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement column renaming in DataFrame, updating both logical names and underlying Frame references.

Browse Source
This commit is contained in:
Palash Tyagi 2025-06-22 05:35:48 +01:00
parent 9b480e8130
commit ff4535c56b
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/dataframe/df.rs
View File

@ -252,9 +252,28 @@ impl DataFrame {
old_name old_name
) )
}); });
self.data.insert(new_name.clone(), column); let new_name_clone = new_name.clone();
self.data.insert(new_name, column);
if let Some(pos) = self.column_names.iter().position(|n| n == old_name) { if let Some(pos) = self.column_names.iter().position(|n| n == old_name) {
self.column_names[pos] = new_name; self.column_names[pos] = new_name_clone.clone();
}
// rename the column in the underlying Frame as well
if let Some(col) = self.data.get_mut(&new_name_clone) {
match col {
DataFrameColumn::F64(frame) => {
frame.rename(old_name, new_name_clone.clone());
}
DataFrameColumn::I64(frame) => {
frame.rename(old_name, new_name_clone.clone());
}
DataFrameColumn::String(frame) => {
frame.rename(old_name, new_name_clone.clone());
}
DataFrameColumn::Bool(frame) => {
frame.rename(old_name, new_name_clone.clone());
}
}
} }
} }